FutureWarning: torch.cpu.amp.autocast(args...) is deprecated. Please use torch.amp.autocast('cpu', args...) instead.

[diana-jung]

Hi Team,

I'm getting the following error while training.
It looks like you need to change torch.cpu.amp.autocast(args...) to torch.amp.autocast('cpu', args...).
It's still in training, but it doesn't seem to be an issue with tuning, and it seems to be a "FutureWarning", like the error log.

Regards,
Diana

[WangChaoYangAm]

就是这个方法以后会弃用，用新的，torch版本换一换，或者更改某个方法，禁用这个日志输出。