Skip to content

Commit 869bac2

Browse files
committed
Fix forward_intermediates() grad_checkpointing in vision_transformer.py
1 parent 2579b12 commit 869bac2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/vision_transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -827,7 +827,7 @@ def forward_intermediates(
827827
if attn_mask is not None:
828828
x = blk(x, attn_mask=attn_mask)
829829
elif self.grad_checkpointing and not torch.jit.is_scripting():
830-
x = checkpoint(blk. x)
830+
x = checkpoint(blk, x)
831831
else:
832832
x = blk(x)
833833
if i in take_indices:

0 commit comments

Comments
 (0)