out of memory的解决方法
收藏
1、减小batch size;
2、减小网络输入的分辨率;
3、通道裁剪;
4、减小网络深度;
5、使用混合精度训练;
6、采用数据分布式训练;
1
收藏
请登录后评论
点赞!