RuntimeError: CUDA out of memory. Tried to allocate 96.00 MiB (GPU 0; 15.75 GiB total capacity; 14.39 GiB already allocated; 51.50 MiB free; 14.71 GiB reserved in total by PyTorch)
把batch_size 设置小一点的数。
RuntimeError: CUDA out of memory. Tried to allocate 96.00 MiB (GPU 0; 15.75 GiB total capacity; 14.39 GiB already allocated; 51.50 MiB free; 14.71 GiB reserved in total by PyTorch)
把batch_size 设置小一点的数。