运行环境：euler_2.8.3
python 3.7.15
atlas910*8 （modelarts环境）
luojianet1.0.6（luojianet1.0.6是基于mindspore1.7改的，可以近似等于mindspore1.7）
在训练DenseNet121网络时，当BatchSize设置较大（例如16，32，64）时，会出现内存溢出，导致训练报错终止。这边通过gitee上看的在mindspore（https://gitee.com/mindspore/models/tree/r1.7/official/cv/densenet）是设置32训练没问题的。代码已经上传能不能麻烦帮忙看下什么去问题
点赞
收藏
回复
分享
举报
运行环境：euler_2.8.3
python 3.7.15
atlas910*8 （modelarts环境）
luojianet1.0.6（luojianet1.0.6是基于mindspore1.7改的，可以近似等于mindspore1.7）
在训练DenseNet121网络时，当BatchSize设置较大（例如16，32，64）时，会出现内存溢出，导致训练报错终止。这边通过gitee上看的在mindspore（https://gitee.com/mindspore/models/tree/r1.7/official/cv/densenet）是设置32训练没问题的。代码已经上传能不能麻烦帮忙看下什么去问题