TypeError: __init__() got an unexpected keyword argument ‘device_type‘

 mark 一个有趣的bug

with autocast(device_type="cuda"):
TypeError: __init__() got an unexpected keyword argument 'device_type'

一般会说 和torch 版本不匹配，

跟它本身没关系

主要是因为

scaler = GradScaler()

这个样子的不匹配

要写这个样的

scaler = torch.amp.GradScaler('cuda')