Torch.cuda.amp vs Nvidia apex?

Hi @Yangmin
You may want to refer to:
https://pytorch.org/docs/stable/notes/amp_examples.html

1 Like