p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change default enable_auto_mixed_precision from true to false

This commit is contained in:
gengdongjie 2020-05-19 17:06:57 +08:00
parent 75f4c598b0
commit a3ed4c6967
4 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
example/resnet101_imagenet2012/train.py
View File

@ -48,7 +48,8 @@ args_opt = parser.parse_args()
device_id = int(os.getenv('DEVICE_ID')) device_id = int(os.getenv('DEVICE_ID'))
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False, device_id=device_id) context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False, device_id=device_id,
enable_auto_mixed_precision=True)
if __name__ == '__main__': if __name__ == '__main__':
if not args_opt.do_eval and args_opt.run_distribute: if not args_opt.do_eval and args_opt.run_distribute:

4
example/resnet50_cifar10/train.py
View File

@ -41,8 +41,8 @@ args_opt = parser.parse_args()
device_id = int(os.getenv('DEVICE_ID')) device_id = int(os.getenv('DEVICE_ID'))
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False) context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False, device_id=device_id,
context.set_context(device_id=device_id) enable_auto_mixed_precision=True)
if __name__ == '__main__': if __name__ == '__main__':
if not args_opt.do_eval and args_opt.run_distribute: if not args_opt.do_eval and args_opt.run_distribute:

5
example/resnet50_imagenet2012/train.py
View File

@ -43,8 +43,9 @@ args_opt = parser.parse_args()
device_id = int(os.getenv('DEVICE_ID')) device_id = int(os.getenv('DEVICE_ID'))
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False)
context.set_context(device_id=device_id) context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False, device_id=device_id,
enable_auto_mixed_precision=True)
if __name__ == '__main__': if __name__ == '__main__':
if not args_opt.do_eval and args_opt.run_distribute: if not args_opt.do_eval and args_opt.run_distribute:

2
mindspore/ccsrc/utils/context/ms_context.cc
View File

@ -72,7 +72,7 @@ MsContext::MsContext(const std::string &policy, const std::string &target) {
enable_mem_reuse_ = true; enable_mem_reuse_ = true;
enable_gpu_summary_ = true; enable_gpu_summary_ = true;
precompile_only_ = false; precompile_only_ = false;
auto_mixed_precision_flag_ = true; auto_mixed_precision_flag_ = false;
enable_pynative_infer_ = false; enable_pynative_infer_ = false;
enable_dynamic_mem_pool_ = true; enable_dynamic_mem_pool_ = true;
graph_memory_max_size_ = "0"; graph_memory_max_size_ = "0";