switch to O2

example/dqn/train.py

@@ -44,6 +44,7 @@ def train(episode=options.episode):
     if context.get_context('device_target') in ['CPU']:
         context.set_context(enable_graph_kernel=True)
     context.set_context(mode=context.GRAPH_MODE, ascend_config={"precision_mode": "allow_mix_precision"})
+    context.set_context(jit_config={"jit_level": "O2"})
     compute_type = mstype.float32 if options.precision_mode == 'fp32' else mstype.float16
     config.algorithm_config['policy_and_network']['params']['compute_type'] = compute_type
     if compute_type == mstype.float16 and options.device_target != 'Ascend':