Fix NLP Demo (#1303)

31cfaf7b · Chang Xu · GitHub · c5b17022 · 31cfaf7b
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

example/auto_compression/nlp/run.py example/auto_compression/nlp/run.py +1 -2

未找到文件。
--- a/example/auto_compression/nlp/run.py
+++ b/example/auto_compression/nlp/run.py
@@ -302,8 +302,7 @@ def main():
        save_dir=args.save_dir,
        config=all_config,
        train_dataloader=train_dataloader,
-        eval_callback=eval_function
-        if 'HyperParameterOptimization' not in all_config else eval_dataloader,
+        eval_callback=eval_function,
        eval_dataloader=eval_dataloader)

    if not os.path.exists(args.save_dir):