未验证 提交 31cfaf7b 编写于 作者: C Chang Xu 提交者: GitHub

Fix NLP Demo (#1303)

上级 c5b17022
...@@ -302,8 +302,7 @@ def main(): ...@@ -302,8 +302,7 @@ def main():
save_dir=args.save_dir, save_dir=args.save_dir,
config=all_config, config=all_config,
train_dataloader=train_dataloader, train_dataloader=train_dataloader,
eval_callback=eval_function eval_callback=eval_function,
if 'HyperParameterOptimization' not in all_config else eval_dataloader,
eval_dataloader=eval_dataloader) eval_dataloader=eval_dataloader)
if not os.path.exists(args.save_dir): if not os.path.exists(args.save_dir):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册