Skip to content

Commit e1ac345

Browse files
Mengjiao Zhoufacebook-github-bot
authored andcommitted
revert D76061842 to unblock IFU UDD conveyor (#3067)
Summary: Pull Request resolved: #3067 Reviewed By: kausv Differential Revision: D76272934 fbshipit-source-id: faf9dd16cfa6b1c7212230467d31204d870df9ee
1 parent d639294 commit e1ac345

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

torchrec/modules/hash_mc_metrics.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,10 @@ def __init__(
4646
) -> None:
4747
super().__init__()
4848

49-
# persist scalar logger steps in checkpoint to make sure it is not reset after training job restarted
5049
self.register_buffer(
5150
ScalarLogger.STEPS_BUFFER,
5251
torch.tensor(1, dtype=torch.int64),
53-
persistent=True,
52+
persistent=False,
5453
)
5554

5655
self._name: str = name

0 commit comments

Comments
 (0)