
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
2024年10月论文阅读--基于强化学习的图像分类
第一篇:2022Reinforcement Learning in Image Classification: AReview(基于强化学习的图像分类综述)作者:目的:关于图像分类在强化学习方面的综述。内容:①通过分析以往的研究结果,说明使用基于rl的方法在图像分类过程中的效果②为RL技术在图像处理和分类中的应用提供未来的方向和开放问题具体:①关于RL的效率有两种指标:数据效率和计算效率。数据效率

运行几轮后报错:torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 62.00 MiB.
能运行十几个epoch之后就报错内存不足。我甚至将batchsize调到了1,能减小的超参数我全调小了,但是依然报这个错。我一度怀疑是梯度的问题,由于梯度一直累计才这样。但是看代码根本没有什么问题。将with torch.no_grad():使用在调用模型前。来确保不保留梯度计算图!于是,在预测时,使用。

到底了