RuntimeError: CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity; 4.53 GiB already allocated; 10.82 MiB free; 4.57 GiB reserved in total by PyTorch)
这个是读取的内容太多,溢出了。把batch_size放小一点。
RuntimeError: CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity; 4.53 GiB already allocated; 10.82 MiB free; 4.57 GiB reserved in total by PyTorch)
这个是读取的内容太多,溢出了。把batch_size放小一点。