We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2d734d9 commit fc3410bCopy full SHA for fc3410b
timm/models/eva.py
@@ -677,7 +677,7 @@ def forward_features(self, x):
677
x, rot_pos_embed = self._pos_embed(x)
678
for blk in self.blocks:
679
if self.grad_checkpointing and not torch.jit.is_scripting():
680
- x = checkpoint(blk, x, rope=rot_pos_embed)
+ x = checkpoint(blk, x, rope=rot_pos_embed, use_reentrant=False)
681
else:
682
x = blk(x, rope=rot_pos_embed)
683
x = self.norm(x)
0 commit comments