GPU问题

[shylie728]

您好，我问请问一下，您说使用2块32GB的GPU来进行训练，但是我使用的是4块10GB的GPU，好像会报错，然后我尝试把batchsize调小，但又会报另外的错，请问这种情况我还可以使用我的服务器进行训练吗？

[Neverland610]

2块32GB的GPU训练能得到比较好训练效果。
如果你要用4块10GB的GPU话

1. 模型效果会比较受限。
2. 你需要调整batch size，不仅是在训练脚本里调，还需要根据报错信息，调整一些代码的写法，因为目前代码里有一些assert是写死的，你需要调整这部分。