diff --git a/train.py b/train.py index 7fa0391..5d6d45a 100755 --- a/train.py +++ b/train.py @@ -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)