记录,pytorch v1.4 似乎有严重bug
pytorch v1.4 cuda101 windows
训练复杂模型时,例如DeepLabV3,收敛会非常困难,猜测是内部梯度计算有错误。
降级到 v1.3.1 cuda101 windows,收敛变的正常
不知道其他版本有没有这样的问题,但如果发现收敛困难,请降级到v1.3.1。
pytorch v1.4 严重收敛bug
最新推荐文章于 2024-07-26 01:14:34 发布
记录,pytorch v1.4 似乎有严重bug
pytorch v1.4 cuda101 windows
训练复杂模型时,例如DeepLabV3,收敛会非常困难,猜测是内部梯度计算有错误。
降级到 v1.3.1 cuda101 windows,收敛变的正常
不知道其他版本有没有这样的问题,但如果发现收敛困难,请降级到v1.3.1。