未验证 提交 892f87e6 编写于 作者: L Lyon 提交者: GitHub

Merge pull request #101 from Oneflow-Inc/close_debug_mode

Close debug mode
......@@ -51,7 +51,7 @@ model_dict = {
flow.config.gpu_device_num(args.gpu_num_per_node)
flow.config.enable_debug_mode(True)
#flow.config.enable_debug_mode(True)
@flow.global_function("predict", get_val_config(args))
def InferenceNet():
assert os.path.exists(args.val_data_dir)
......
......@@ -50,7 +50,7 @@ model_dict = {
flow.config.gpu_device_num(args.gpu_num_per_node)
flow.config.enable_debug_mode(True)
#flow.config.enable_debug_mode(True)
def label_smoothing(labels, classes, eta, dtype):
......
......@@ -134,7 +134,7 @@ if __name__ == "__main__":
configs.print_args(args)
flow.config.gpu_device_num(args.gpu_num_per_node)
flow.config.enable_debug_mode(True)
#flow.config.enable_debug_mode(True)
@flow.global_function(get_val_config(args))
def IOTest():
if args.train_data_dir:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册