Apex with `torch.optim.swa_utils`

distributed
2

apex.amp is deprecated in favor of the native torch.cuda.amp implementation and we recommend to switch to the latter.
More details are given in this post.