Pytorch warning with autocast
pfeatherstone opened this issue · 1 comments
pfeatherstone commented
With latest Pytorch version i'm getting the warning:
FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead
lucidrains commented
how far back in torch versions does this work?