src/transformers/models/audio_spectrogram_transformer/modeling_audio_spectrogram_transformer.py/ASTPreTrainedModel/_set_gradient_checkpointing
class ASTPreTrainedModel:    def _set_gradient_checkpointing(self, module: ASTEncoder, value: bool = False) -> None:
        if isinstance(module, ASTEncoder):
            module.gradient_checkpointing = value