[Fix] Fix the bug that uses wrong dataloader while evaluating.

chenyaofo · Nov 30, 2021 · 9014bb4 · 9014bb4
1 parent 81e70a9
commit 9014bb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/codebase/main.py b/codebase/main.py
@@ -60,7 +60,7 @@ def excute_pipeline(
         metric_store += evaluate_one_epoch(
             epoch=0,
             model=model,
-            loader=train_loader,
+            loader=val_loader,
             criterion=criterion,
             optimizer=optimizer,
             scheduler=scheduler,