diff --git a/fbgemm_gpu/fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py b/fbgemm_gpu/fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py index 0f9b4346c..3d2f24a93 100644 --- a/fbgemm_gpu/fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py +++ b/fbgemm_gpu/fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py @@ -48,7 +48,7 @@ def __init__(self, wrapped) -> None: wrapped: torch.classes.fbgemm.KVTensorWrapper """ self._wrapped = wrapped - self._requires_grad = True + self._requires_grad = False @property def wrapped(self):