Hello @JosephC thanks for your answer ! But if I run model.eval() I will not be allowed to put the target in the command model(images, targets).
Also If you know how to do it, I would like during the training mode to get the prediction and the loss. But when I do model(images, targets) I only get the loss.
Thanks for your help !