diff --git a/train.py b/train.py index 42b9bb0..013fe33 100644 --- a/train.py +++ b/train.py @@ -365,6 +365,7 @@ def val(args, val_loader, model): with torch.no_grad(): # input_var = Variable(input).cuda() input_var = input.cuda() + start_time = time.time() output = model(input_var) time_taken = time.time() - start_time print("[%d/%d] time: %.2f" % (i + 1, total_batches, time_taken))