qq_41958789的博客

原创 复现nnUnetv2遇到的一个错误

摘要：文章指出错误源于self.grad_scaler.scale(l).backward()调用，因GradScaler初始化参数错误。错误原因是将字符串"cuda"作为参数传入GradScaler构造函数，而该构造函数不支持字符串参数。解决方法是将初始化语句self.grad_scaler=GradScaler("cuda")改为self.grad_scaler=GradScaler()，保留条件判断self.device.type=='cuda'。