Skip to content

Commit 922f33f

Browse files
fix train
1 parent 630cb60 commit 922f33f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def main(args):
9898
# 2. validate with validate dataset
9999
if config.validate and epoch_id % config.valid_interval == 0:
100100
net.eval()
101-
top1_acc = program.run(valid_dataloader, config, net, None,
101+
top1_acc = program.run(valid_dataloader, config, net, None, None,
102102
epoch_id, 'valid')
103103
if top1_acc > best_top1_acc:
104104
best_top1_acc = top1_acc

0 commit comments

Comments
 (0)