未验证 提交 e33f641b 编写于 作者: T Teng Xi 提交者: GitHub

add Paddle 1.8 PaddleSlim 1.1.1 demo (#304)

* add Paddle 1.8 PaddleSlim 1.1.1 demo
上级 5b92f265
......@@ -2,6 +2,8 @@
本示例将演示如何训练`slimfacenet`及评测`slimfacenet`量化模型。
本示例依赖:Paddle 1.8 PaddleSlim 1.1.1
当前示例支持以下人脸识别模型:
- `SlimFaceNet_A_x0_60`
......
......@@ -233,9 +233,7 @@ def quant_val_reader_batch():
test_dataset = LFW(nl, nr)
test_reader = paddle.fluid.io.batch(
test_dataset.reader, batch_size=1, drop_last=False)
shuffle_index = args.seed if args.seed else np.random.randint(1000)
print('shuffle_index: {}'.format(shuffle_index))
shuffle_reader = fluid.io.shuffle(test_reader, shuffle_index)
shuffle_reader = fluid.io.shuffle(test_reader, 3)
def _reader():
while True:
......@@ -289,7 +287,6 @@ def main():
'--start_epoch', default=0, type=int, help='start_epoch')
parser.add_argument(
'--total_epoch', default=80, type=int, help='total_epoch')
parser.add_argument('--seed', default=None, type=int, help='shuffle seed')
parser.add_argument(
'--save_frequency', default=1, type=int, help='save_frequency')
parser.add_argument(
......@@ -336,8 +333,10 @@ def main():
sample_generator=quant_val_reader_batch(),
model_filename=None, #'model',
params_filename=None, #'params',
batch_size=100,
batch_nums=10)
save_model_filename=None, #'model',
save_params_filename=None, #'params',
batch_size=np.random.randint(80, 160),
batch_nums=np.random.randint(4, 10))
elif args.action == 'test':
[inference_program, feed_target_names,
fetch_targets] = fluid.io.load_inference_model(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册