未验证 提交 60e1ff60 编写于 作者: B Bin Lu 提交者: GitHub

remove gpu infer (#5530)

* remove gpu inference
上级 3ce9dd26
......@@ -25,7 +25,7 @@ train_model:./pretrain_models/mobilenet_v3_small_pretrained.pdparams
infer_export:tools/export_model.py --model=mobilenet_v3_small
##
inference:deploy/inference_python/infer.py
--use-gpu:True|False
--use-gpu:False
--batch-size:1
--model-dir:./output/mobilenet_v3_small_infer/
--img-path:./images/demo.jpg
......
......@@ -281,6 +281,7 @@ def main(args):
train_one_epoch(model, criterion, optimizer, data_loader, epoch,
args.print_freq, args.amp_level, scaler)
lr_scheduler.step()
if paddle.distributed.get_rank() == 0:
top1 = evaluate(
model, criterion, data_loader_test, amp_level=args.amp_level)
if args.output_dir:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册