提交 a72650a5 编写于 作者: S ShawnXuan 提交者: mir-of

change: function->global_function

上级 b835016d
...@@ -35,7 +35,7 @@ flow.config.gpu_device_num(args.gpu_num_per_node) ...@@ -35,7 +35,7 @@ flow.config.gpu_device_num(args.gpu_num_per_node)
flow.config.enable_debug_mode(True) flow.config.enable_debug_mode(True)
@flow.function(get_val_config(args)) @flow.global_function(get_val_config(args))
def InferenceNet(): def InferenceNet():
if args.val_data_dir: if args.val_data_dir:
assert os.path.exists(args.val_data_dir) assert os.path.exists(args.val_data_dir)
......
...@@ -123,7 +123,7 @@ if __name__ == "__main__": ...@@ -123,7 +123,7 @@ if __name__ == "__main__":
flow.config.gpu_device_num(args.gpu_num_per_node) flow.config.gpu_device_num(args.gpu_num_per_node)
flow.config.enable_debug_mode(True) flow.config.enable_debug_mode(True)
@flow.function(get_val_config(args)) @flow.global_function(get_val_config(args))
def IOTest(): def IOTest():
if args.train_data_dir: if args.train_data_dir:
assert os.path.exists(args.train_data_dir) assert os.path.exists(args.train_data_dir)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册