We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4612086 commit aed7e96Copy full SHA for aed7e96
src/super_gradients/training/utils/weight_averaging_utils.py
@@ -49,7 +49,7 @@ def __init__(
49
else:
50
averaging_snapshots_dict["snapshots_metric"] = np.inf * np.ones(self.number_of_models_to_average)
51
52
- torch.save(averaging_snapshots_dict, self.averaging_snapshots_file)
+ torch.save(averaging_snapshots_dict, self.averaging_snapshots_file)
53
54
def update_snapshots_dict(self, model, validation_results_tuple):
55
"""
0 commit comments