scaler

[Benbie]

assert len(optimizer_state["found_inf_per_device"]) > 0, "No inf checks were recorded for this optimizer."
作者大大您好，我在三维医学图像上用您的方法，scaler报错这个，去掉scaler后loss完全不变，参数没有更新，请问可能是什么问题

[Haochen-Wang409]

您具体是怎么“去掉”scaler的？修改了哪些代码？

[Benbie]

我将这部分换成了传统的 loss.backward() optimizer.step()

[Benbie]

然后loss_learn保持不变，可能是参数完全没有任何更新

[Haochen-Wang409]

在其他的数据集，如ImageNet上，可以正常运行吗？

[Benbie]

我将试一下其他的三维数据集，谢谢您

[Haochen-Wang409]

建议您在ImageNet数据集先跑通，因为我们并没有在其他数据集上进行预训练。