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 6e93e77 commit 3d5a9f6Copy full SHA for 3d5a9f6
torchrec/distributed/embeddingbag.py
@@ -597,10 +597,7 @@ def __init__(
597
if env.process_group and dist.get_backend(env.process_group) != "fake":
598
self._initialize_torch_state()
599
600
- if module.device not in ["meta", "cpu"] and module.device.type not in [
601
- "meta",
602
- "cpu",
603
- ]:
+ if module.device != "meta" and module.device.type != "meta":
604
self.load_state_dict(module.state_dict(), strict=False)
605
606
@classmethod
0 commit comments