Skip to content

Commit aed7e96

Browse files
authored
fixed save in SWA (#1111)
1 parent 4612086 commit aed7e96

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/super_gradients/training/utils/weight_averaging_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def __init__(
4949
else:
5050
averaging_snapshots_dict["snapshots_metric"] = np.inf * np.ones(self.number_of_models_to_average)
5151

52-
torch.save(averaging_snapshots_dict, self.averaging_snapshots_file)
52+
torch.save(averaging_snapshots_dict, self.averaging_snapshots_file)
5353

5454
def update_snapshots_dict(self, model, validation_results_tuple):
5555
"""

0 commit comments

Comments
 (0)