Skip to content

model_transform has been added to create_supervised_trainer and create_supervised_evaluator but its not used by create_supervised_evaluator #2894

@david-waterworth

Description

@david-waterworth

🐛 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.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions