CUDA out of memory

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放小一点。

你可能感兴趣的:(自然语言处理,pytorch)