This commit is contained in:
Glenn Jocher 2019-12-22 13:04:05 -08:00
parent 5766b5c555
commit 8a5c520291

View File

@ -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