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 24bb848 commit bfbd9aeCopy full SHA for bfbd9ae
torchrec/modules/hash_mc_modules.py
@@ -405,6 +405,8 @@ def _load_state_dict_pre_hook(
405
return
406
if "_hash_zch_metadata" in state_dict:
407
del state_dict["_hash_zch_metadata"]
408
+ if "_hash_zch_bucket" in state_dict:
409
+ del state_dict["_hash_zch_bucket"]
410
411
self._register_load_state_dict_pre_hook(
412
_load_state_dict_pre_hook, with_module=True
0 commit comments