训练的时候给的batch为8能运行，测试的时候batch为1，显存都不够用呢？

[LaiPiXiong]

32G显卡，batch为1都带不动。。。。。。。

[Fulep]

你好，可以参考下你的训练代码吗？非常感谢！！！[email protected]

[msn199959]

32G显卡，batch为1都带不动。。。。。。。

用GitHub上BalancedDataParallel接口可以解决

[wangrui9]

32G显卡，batch为1都带不动。。。。。。。

用GitHub上BalancedDataParallel接口可以解决

我最近也被这个bs为1显存不够困扰着，我了解了一下BalancedDataParallel，这个是用来平衡多个gpu运算的，bs已经为1了，我认为再使用这个BalancedDataParallel也是没有意义的

[zuofengyouliang]

您好，能参考一下train的代码吗，万分感谢！！！[email protected]

[CarpeDiemly]

您好，请问可以参考一下您的代码吗，主要是PRA Loss部分，万分感谢！[email protected]

[18339508629]

您好，能参考一下train的代码吗，万分感谢！[email protected]

[cyn2399]

您好，能参考一下您的train.py的代码吗？或者发送到[email protected]

[wangrui9]

你好，以下是关于内推申请的说明，如还有其他问题请添加qq 1406363544同学你好，我只是普通员工，本职工作不是收简历，我还有其他工作要做，最近邮箱收到的简历太多了，实在是内推不过来了，没法靠我个人将这么多份简历填到内网上，尤其很多同学邮件内容还有岗位和意向地不匹配等问题；请去我们公司的校招 官网/公众号 投递，或者联系我们公司的其他员工内推吧，个人精力有限，带来不便敬请谅解