🐛 Bug description
#2848 added model_transform to both create_supervised_trainer and create_supervised_evaluator but its not used by create_supervised_evaluator
|
def create_supervised_evaluator( |
For consistency it should be implemented in both (i.e. supervised_evaluation_step and supervised_evaluation_step_amp)- otherwise, evaluation fails if the model returns a dictionary for example.