2023年12月14日
pytorch减少显存方式
摘要： 引导 1. 显存都用在哪儿了？ 2. 技巧 1：使用就地操作 3. 技巧 2：避免中间变量 4. 技巧 3：优化网络模型 5. 技巧 4：减小 BATCH_SIZE 6. 技巧 5：拆分 BATCH 7. 技巧 6：降低 PATCH_SIZE 8. 技巧 7：优化损失求和 9. 技巧 8：调整训练精 阅读全文
posted @ 2023-12-14 17:26 海_纳百川 阅读(418) 评论(2) 推荐(1) 编辑
本站总访问量次