PyTorch:日常使用笔记及踩坑

it2024-11-05  16

1. 半精度Nan

参考资料:https://blog.csdn.net/one_six_mix/article/details/86367086 设置优化器eps参数,保证除法不出现除0操作。

optim.Adam(model.parameters(), lr=1e-3, eps=1e-4)
最新回复(0)