open-mmlab/mmsegmentation

Too much GPU memory consumption and seems wrong

Open

#3,167 建立於 2023年7月6日

在 GitHub 查看
 (18 留言) (0 反應) (0 負責人)Python (2,386 fork)batch import
good first issue

倉庫指標

Star
 (6,867 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

When I use resnet101 with deeplabv3+ in this lib 'https://github.com/VainF/DeepLabV3Plus-Pytorch', I can train for 40 batch size on 2 GPUs, 20 batch size for each GPU, and it consumes about 36 GB. However, when I use your lib, for 2 batch size on 1 GPUs, it consumes about 15 GB

貢獻者指南

研究方向
使用 nvidia smi 或 torch.cuda.memory summary 等工具分析 GPU 記憶體使用情況，與 DeepLabV3Plus Pytorch 比較，找出張量分配或梯度檢查點方面的差異。
技術棧
pythonpytorch
領域
machine learningbackend
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-2 天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
大致清晰
前置要求
PythonPyTorchGPU memory profiling
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
40