Torch.cuda.amp vs Nvidia apex? - PyTorch Forums thumbnail
Torch.cuda.amp vs Nvidia apex? - PyTorch Forums
discuss.pytorch.org
Yes, apex.amp was our first implementation of mixed-precision training, is deprecated now, and replaced with torch.cuda.amp. @seungjun posted the examples above to see its usage.
1 Users
0 Comments
1 Highlights
0 Notes

Top Highlights

  • Yes, apex.amp was our first implementation of mixed-precision training, is deprecated now, and replaced with torch.cuda.amp. @seungjun posted the examples above to see its usage.

Ready to highlight and find good content?

Glasp is a social web highlighter that people can highlight and organize quotes and thoughts from the web, and access other like-minded people’s learning.