This version still uses apex (deprecated) for mixed precision. In the next release, it will be updated with the pytorch build-in feature torch.autocast and torch.cuda.amp.GradScaler.
This version still uses apex (deprecated) for mixed precision. In the next release, it will be updated with the pytorch build-in feature torch.autocast and torch.cuda.amp.GradScaler.