Skip to content

Commit

Permalink
Quick fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Towaki Takikawa authored and Towaki Takikawa committed Oct 8, 2019
1 parent 38a278d commit 12f5cfd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ def main():

if args.evaluate:
# Early evaluation for benchmarking
default_eval_epoch = 1
default_eval_epoch = 1
validate(val_loader, net, criterion_val,
optim, default_eval_epoch, writer)
evaluate(val_loader, net)
Expand Down

0 comments on commit 12f5cfd

Please sign in to comment.