Apex AMP with torch.cuda.amp

mixed-precision
4

apex.amp is deprecated so you should stick to the native implementation. This post explains it in more detail.

1 Like