出现类似的错误,试着改小bath_size的值试试。
RuntimeError: CUDA out of memory. Tried to allocate 352.00 MiB (GPU 0; 7.80 GiB total capacity; 6.45
最新推荐文章于 2024-02-21 07:55:16 发布
出现类似的错误,试着改小bath_size的值试试。