mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix #1935 , torch.amp.autocast -> torch.autocast (namespace issue for 1.10 pt compat)

Browse Source
This commit is contained in:
Ross Wightman 2023-08-30 15:02:20 -07:00
parent 9260cf517d
commit 983310d6a2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientvit_mit.py
View File

@ -260,7 +260,7 @@ class LiteMSA(nn.Module):
v = F.pad(v, (0, 1), mode="constant", value=1.)
if not torch.jit.is_scripting():
with torch.amp.autocast(device_type=v.device.type, enabled=False):
with torch.autocast(device_type=v.device.type, enabled=False):
out = self._attn(q, k, v)
else:
out = self._attn(q, k, v)