(module)
| 897 | if self.training and self.gradient_checkpointing: |
| 898 | |
| 899 | def create_custom_forward(module): |
| 900 | def custom_forward(*inputs): |
| 901 | return module(*inputs) |
| 902 | |
| 903 | return custom_forward |
| 904 | |
| 905 | hidden_states = torch.utils.checkpoint.checkpoint( |
| 906 | create_custom_forward(resnet), hidden_states, temb |
nothing calls this directly
no outgoing calls
no test coverage detected