Is your feature request related to a problem? Please describe.
provide an option for all the evaluators to be able to switch between "train_mode" and "eval_mode" of the model.
(this is useful for GradCAM inference integration)

Additional context