问题解决：RuntimeError: CUDA out of memory.(....; 5.83 GiB reserved in total by PyTorch)

https://blog.csdn.net/weixin_41587491/article/details/105488239
可以改batch_size 通常有64、32啥的