diff --git a/test.py b/test.py index 5a85bbe4..bdcf8cf5 100644 --- a/test.py +++ b/test.py @@ -80,7 +80,6 @@ def test(cfg, # Disable gradients with torch.no_grad(): - # Run model inf_out, train_out = model(imgs) # inference and training outputs