Skip to content

task eval broken for PyTorch engine #1779

Description

@Gerstenberger

The task eval is currently broken for the backend=torch.

The call

returnn/returnn/__main__.py

Lines 563 to 570 in e1762d8

engine.eval_model(
output_file=config.value("eval_output_file", None),
output_per_seq_file=config.value("eval_output_file_per_seq", None),
loss_name=config.value("loss_name", None),
output_per_seq_format=config.list("output_per_seq_format", ["score"]),
output_per_seq_file_format=config.value("output_per_seq_file_format", "txt"),
lr_control_update_scores=lr_control_update_scores,
)

is incompatible with the torch engine's signature of

def eval_model(self, *, skip_already_evaluated: bool = False):

Is the eval task intended to be used with PyTorch? I don't see a particular reason why it shouldn't.
I have a use case where i want to compare the output of a forward job with a callback with the output of a eval job, for which I know the loss is numerically correct.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions