update autocast - Githubissues

lucidrains / vector-quantize-pytorch

Vector (and Scalar) Quantization, in Pytorch

MIT License

2.65k stars 215 forks source link

Closed al-jshen closed 2 months ago

al-jshen commented 3 months ago

from torch: torch.cuda.amp.autocast(args...) is deprecated. Please use torch.amp.autocast('cuda', args...) instead.

lucidrains commented 3 months ago

@al-jshen how far back in torch versions does torch.amp.autocast('cuda', ...) work for?