求教

[meox3259]

想问一下这个真的跑得起来吗，为啥我跑了下中间显存显示要2000个G

[h-z-h-cell]

我在我本地电脑上可以跑cifa10和cifar10dvs那两个，imagenet我没试过。

[meox3259]

我用imagenet能跑通，但是cifar不太行...

[meox3259]

想问下是直接用模型跑就行吗，不需要修改

[h-z-h-cell]

我电脑上就是直接用这个代码这么跑的，只不过数据集我没有放上来，需要自己下载。我设置batch_size为128的时候会提示我内存60G不够用，我换成batch_size为64才能跑，如果跑不动，试试要不要再缩小一点batch_size。

[meox3259]

感谢