From d9551a20190b3fc244a4279ddfb5136fbdd81c58 Mon Sep 17 00:00:00 2001 From: Sander van Leeuwen Date: Sat, 1 May 2021 15:34:29 +0200 Subject: [PATCH] Back out temporary code --- validate.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/validate.py b/validate.py index 3f2665fc..a49eac9e 100755 --- a/validate.py +++ b/validate.py @@ -235,15 +235,6 @@ def validate(args): if real_labels is not None: real_labels.add_result(output) - print(input.shape) - print(target.shape) - print(output.shape) - import numpy as np - max_idx = np.argmax(output[0].cpu().detach().numpy()) - - print('Output: ', max_idx, output[0][max_idx], ' should be ', target[0]) - print(output[0]) - exit() # measure accuracy and record loss acc1, acc5 = accuracy(output.detach(), target, topk=(1, 5)) losses.update(loss.item(), input.size(0))