一、实验设置
用自己的dataset进行十分类的任务
二、bug
RuntimeError: CUDA out of memory. Tried to allocate … MiB
三、解决方法
1.调小batch_size–>严重影响实验结果和模型性能
2.使用os.environ[ “CUDA_VISIBLE_DEVICES” ] = '5’指定GPU训练,而不是使用默认GPU0–>成功
bug记录:RuntimeError: CUDA out of memory. Tried to allocate ... MiB
最新推荐文章于 2024-06-14 16:03:35 发布