enable critic_gradient_checkpointing, get error

[BaiStone2017]

self.critic.gradient_checkpointing_enable() File "/opt/conda/lib/python3.8/site-packages/deepspeed/runtime/engine.py", line 454, in __getattr__ raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'") AttributeError: 'DeepSpeedEngine' object has no attribute 'gradient_checkpointing_enable'
how to solve this problem?

[iamsile]

Does anyone know how to fix it ?

[yxzero]

def gradient_checkpointing_enable(self):
self.rwtranrsformer.gradient_checkpointing_enable()
def gradient_checkpointing_disable(self):
self.rwtranrsformer.gradient_checkpointing_disable()

add it in your reward model

[iamsile]

def gradient_checkpointing_enable(self): self.rwtranrsformer.gradient_checkpointing_enable() def gradient_checkpointing_disable(self): self.rwtranrsformer.gradient_checkpointing_disable()

add it in your reward model

yes, it works