diff --git a/example/yolov3_coco2017/eval.py b/example/yolov3_coco2017/eval.py index 8347f5be37..f85fa95d77 100644 --- a/example/yolov3_coco2017/eval.py +++ b/example/yolov3_coco2017/eval.py @@ -82,8 +82,7 @@ if __name__ == '__main__': args_opt = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", device_id=args_opt.device_id) - context.set_context(enable_task_sink=True, enable_loop_sink=True, enable_mem_reuse=True, - enable_auto_mixed_precision=False) + context.set_context(enable_task_sink=True, enable_loop_sink=True, enable_mem_reuse=True) # It will generate mindrecord file in args_opt.mindrecord_dir, # and the file name is yolo.mindrecord0, 1, ... file_num. diff --git a/example/yolov3_coco2017/train.py b/example/yolov3_coco2017/train.py index bccc66d996..6a9729f200 100644 --- a/example/yolov3_coco2017/train.py +++ b/example/yolov3_coco2017/train.py @@ -85,8 +85,7 @@ if __name__ == '__main__': args_opt = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", device_id=args_opt.device_id) - context.set_context(enable_task_sink=True, enable_loop_sink=True, enable_mem_reuse=True, - enable_auto_mixed_precision=False) + context.set_context(enable_task_sink=True, enable_loop_sink=True, enable_mem_reuse=True) if args_opt.distribute: device_num = args_opt.device_num context.reset_auto_parallel_context()