kronfluence icon indicating copy to clipboard operation
kronfluence copied to clipboard

AMP warning

Open pomonam opened this issue 1 year ago • 0 comments

On the latest version, we get the warning message:

FutureWarning: `torch.cuda.amp.GradScaler(args...)` is deprecated. Please use `torch.amp.GradScaler('cuda', args...)` instead.

pomonam avatar Aug 05 '24 01:08 pomonam