fix

026c9f51 · shippingwang · 20afb50a · 026c9f51
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tools/train.py tools/train.py +1 -1

未找到文件。
--- a/tools/train.py
+++ b/tools/train.py
@@ -133,7 +133,7 @@ def main(args):

                top1_acc = program.run(valid_dataloader, exe,
                                       compiled_valid_prog, valid_fetchs,
-                                       epoch_id, 'valid')
+                                       epoch_id, 'valid', config)
                if top1_acc > best_top1_acc:
                    best_top1_acc = top1_acc
                    message = "The best top1 acc {:.5f}, in epoch: {:d}".format(