Skip to content
Snippets Groups Projects
Commit ac5c9806 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[Trainer] changed the invalid index of a 0-dim tensor (data[0] to item())

parent 5498ea23
Branches
Tags
No related merge requests found
Pipeline #
......@@ -169,8 +169,8 @@ class CNNTrainer(object):
optimizer.step()
end = time.time()
logger.info("[{}/{}][{}/{}] => Loss = {} (time spent: {})".format(epoch, n_epochs, i, len(dataloader), loss.data[0], (end-start)))
losses.append(loss.data[0])
logger.info("[{}/{}][{}/{}] => Loss = {} (time spent: {})".format(epoch, n_epochs, i, len(dataloader), loss.item(), (end-start)))
losses.append(loss.item())
# do stuff - like saving models
logger.info("EPOCH {} DONE")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment