请问使用vision.models.MobileNetV2爆显存怎么解决

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

厌食的贪吃蛇发布于2022-05

我使用vision.models.MobileNetV2每次迭代显存会累积，然后就爆显存了，请问如何解决？大概每次会增加8个G左右

0

收藏

回复

全部评论(3)

beyondyourself

#2 回复于2022-05

缓存清理添加一下

0

回复

李长安

#4 回复于2022-07

是不是保存了什么东西？loss之类的，看看吧

0

回复

厌食的贪吃蛇

#5 回复于2022-07

谢谢解决了推理的时候忘记resize了

0

回复

提issue

需求/bug反馈？一键提issue告诉我们

提pr

发现bug？如果您知道修复办法，欢迎提pr直接参与建设飞桨~