Skip to content

Commit 29755d8

Browse files
author
pytorchbot
committed
2025-01-16 nightly release (4685bac)
1 parent 6b8ecdd commit 29755d8

File tree

4 files changed

+337
-245
lines changed

4 files changed

+337
-245
lines changed

torchrec/distributed/mc_modules.py

-1
Original file line numberDiff line numberDiff line change
@@ -812,7 +812,6 @@ def sharded_parameter_names(self, prefix: str = "") -> Iterator[str]:
812812
if name in [
813813
"_output_segments_tensor",
814814
"_current_iter_tensor",
815-
"_scalar_logger._scalar_logger_steps",
816815
]:
817816
continue
818817
if name in module._non_persistent_buffers_set:

0 commit comments

Comments
 (0)