-
Prabhat Roy authored
* Updated classification reference script to use torch.cuda.amp * Assigned scaler to None if amp is False * Fixed linter errors
403bded3
* Updated classification reference script to use torch.cuda.amp * Assigned scaler to None if amp is False * Fixed linter errors