Skip to content

Commit

Permalink
Merge pull request #42 from eltociear/patch-1
Browse files Browse the repository at this point in the history
chore: update base.py
  • Loading branch information
roman-bushuiev authored Nov 9, 2024
2 parents 2e0ac4a + 6f108e8 commit bd6dc18
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion massspecgym/models/simulation/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ def on_batch_end(
"""
Compute evaluation metrics for the retrieval model based on the batch and corresponding predictions.
This method will be used in the `on_train_batch_end`, `on_validation_batch_end`, since `on_test_batch_end` is
overriden below.
overridden below.
"""
self.evaluate_cos_similarity_step(
outputs["spec_pred"],
Expand Down

0 comments on commit bd6dc18

Please sign in to comment.