我尝试使用pytorch运行深度学习模型。 这是模型链接： https://github.com/JiaRenChang/PSMNet https://github.com/KinglittleQ/PSMNet
在这两个模型中，我遇到相同的问题。 我使用了4个taitan xp GPU，但是模型仅在 batchsize设置为1 时才能开始训练，而当我将batchsize设置为其他数字（偶数2）时，它将在第一个训练时期之前卡住。
我的测试照片显示它将在运行第一个卷积层之前停止。
shangxinde86 回答：pytorch模型只能在批量大小等于1时启动火车吗？
暂时没有好的解决方案，如果你有好的解决方案，请发邮件至：iooj@foxmail.com