GPU使用情况

[heidapao2b]

首先感谢您的出色工作，我想请问该网络训练大约需要多大的GPU内存？

[Chopper-233]

我们是在8张V100上训练的，batchsize是32，如果把batchsize设置为1，只用1张卡（大约32GB)，应该也是可以跑的，但可能会出现不收敛或者效果不够好的情况

[zhiqing0205]

Although there are only four samples per class in the training set, why can the batch size be increased to four?