Thank you @emcp
Sorry, I did see this tutorial. My training procedure is similar to this one.
However, in this tutorial they don’t calculate both loss and IoU for evaluation- only IoU.
I have also looked at their evaluate() function which runs over test_loader, but it only calculates IoU and AP, not loss.
I want to also calculate the loss on the test_loader in order to make sure that I don’t overfit over train_loader while training.
Thank you