kronfluence
kronfluence copied to clipboard
AMP warning
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.