Skip to content

Commit 9014bb4

Browse files
author
chenyaofo
committed
[Fix] Fix the bug that uses wrong dataloader while evaluating.
1 parent 81e70a9 commit 9014bb4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

codebase/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def excute_pipeline(
6060
metric_store += evaluate_one_epoch(
6161
epoch=0,
6262
model=model,
63-
loader=train_loader,
63+
loader=val_loader,
6464
criterion=criterion,
6565
optimizer=optimizer,
6666
scheduler=scheduler,

0 commit comments

Comments
 (0)