Skip to content

Conversation

@Lokkuchakreshkumar
Copy link

The metadata dictionary in TorchExperiment._evaluate() was always empty. This fix populates it with useful training information:

  • num_epochs_trained: The number of epochs the model was trained for
  • all_metrics: All metrics collected during training

This is consistent with other Hyperactive integrations (sklearn, sktime) which also return useful metadata.

…e() method

The metadata dictionary in TorchExperiment._evaluate() was always empty. This fix populates it with useful training information:
- num_epochs_trained: The number of epochs the model was trained for
- all_metrics: All metrics collected during training

This is consistent with other Hyperactive integrations (sklearn, sktime) which also return useful metadata.
Copy link
Collaborator

@SimonBlanke SimonBlanke left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there a way to test this, so that we catch problems like this in the future?

You should also mention the issue #213 and ask for assignment to the issue beforehand.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants