Closed xiaero closed 11 months ago
加载预训练模型的失败 每次到 46/3510 就会停止 我的环境 Python 3.8 cuda 1.8 pytorch 2.0.1 easydict 1.9 einops 0.7.0
把--sample_size 50换成--sample_size all就好啦
--sample_size 50
--sample_size all
谢谢 ,刚看了一下 代码 看到了 因为那取整计算那块 让我一直以为50个失败了 一直卡46就停止了
加载预训练模型的失败 每次到 46/3510 就会停止 我的环境 Python 3.8 cuda 1.8 pytorch 2.0.1 easydict 1.9 einops 0.7.0