comment cudnn_conv_enable_pseudo_half

cnn_e2e/job_function_util.py

@@ -14,8 +14,8 @@ def _default_config(args):
 def get_train_config(args):
     train_config = _default_config(args)
     train_config.train.primary_lr(args.learning_rate)
-    train_config.disable_all_reduce_sequence(True)
-    train_config.cudnn_conv_enable_pseudo_half(True)
+    train_config.disable_all_reduce_sequence(False)
+    #train_config.cudnn_conv_enable_pseudo_half(True)
     #train_config.all_reduce_group_min_mbyte(8)
     #train_config.all_reduce_group_num(128)
     # train_config.all_reduce_lazy_ratio(0)
@@ -35,4 +35,5 @@ def get_train_config(args):
     return train_config
 
 def get_val_config(args):
-    return _default_config(args)
\ No newline at end of file
+    return _default_config(args)
+