Hello .
How can I make my model output different results while model is in train mode and while model is in eval.
For instance let’s say during train I would want my model to output the loss while during eval I would want my model to directly output the results.
I was looking for the functionality that present in the fasterrcnn_resnet50_fpn model wherein
The behavior of the model changes depending if it is in training or evaluation mode.