请教

[pipijiev12]

我想问一下 你这个程序运行gpu大概是多大 因为我gpu才只有6g跑不起来

[jefferyYu]

印象中应该是11GB左右。如果跑不起来，可以把batch_size设置小一点。