src/transformers/models/unispeech_sat/modeling_unispeech_sat.py/UniSpeechSatPreTrainedModel/_set_gradient_checkpointing
class UniSpeechSatPreTrainedModel:    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, (UniSpeechSatEncoder, UniSpeechSatEncoderStableLayerNorm, UniSpeechSatFeatureEncoder)):
            module.gradient_checkpointing = value