RuntimeError: CUDA out of memory. Tried to allocate 352.00 MiB (GPU 0; 7.80 GiB total capacity; 6.45

出现类似的错误，试着改小bath_size的值试试。

版权声明：本文为iamjingong原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。