We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d639294 commit e1ac345Copy full SHA for e1ac345
torchrec/modules/hash_mc_metrics.py
@@ -46,11 +46,10 @@ def __init__(
46
) -> None:
47
super().__init__()
48
49
- # persist scalar logger steps in checkpoint to make sure it is not reset after training job restarted
50
self.register_buffer(
51
ScalarLogger.STEPS_BUFFER,
52
torch.tensor(1, dtype=torch.int64),
53
- persistent=True,
+ persistent=False,
54
)
55
56
self._name: str = name
0 commit comments