(self, enable=True)
| 1030 | |
| 1031 | @torch.jit.ignore |
| 1032 | def set_grad_checkpointing(self, enable=True): |
| 1033 | visual = self.image_encoder_without_ddp.visual |
| 1034 | transformer = self.text_encoder_without_ddp.transformer |
| 1035 | if hasattr(visual, 'set_grad_checkpointing'): |
| 1036 | visual.set_grad_checkpointing(enable) |
| 1037 | if transformer is not None and hasattr(transformer, 'set_grad_checkpointing'): |
| 1038 | transformer.set_grad_checkpointing(enable) |
| 1039 | |
| 1040 | def image_named_params(self): |
| 1041 | return self._image_encoder.named_parameters() |
no outgoing calls