Fix the bug that uses wrong dataloader while evaluation.

chenyaofo · chenyaofo · commit 19a5ce630de8 · 2021-12-01T14:59:11.000+08:00
diff --git a/codebase/main.py b/codebase/main.py
@@ -97,7 +97,7 @@ def excute_pipeline(
         metric_store += evaluate_one_epoch(
             epoch=epoch,
             model=model,
-            loader=train_loader,
+            loader=val_loader,
             criterion=criterion,
             optimizer=optimizer,
             scheduler=scheduler,
