|
|
|
@ -41,8 +41,8 @@ args_opt = parser.parse_args()
|
|
|
|
|
|
|
|
|
|
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 not args_opt.do_eval and args_opt.run_distribute:
|
|
|
|
|