RVC-Project/Retrieval-based-Voice-Conversion-WebUI
在 GitHub 查看MacOS | MPS | torch metal work on Train -> step-1 then Step-2 but not on training the model (Step-3)
Open
#991 建立於 2023年8月12日
help wanted
倉庫指標
- Star
- (18,427 star)
- PR 合併指標
- (30 天內沒有已合併 PR)
描述
Hi is there a way to make the model train on torch metal, mps, even when passing the "mps" manually to the G and D models, it still uses the cpu, I also monitored the GPU performence, it peaks only on the first two steps, but on the model training it peaks the CPU any idea? and it stuck sometimes on "INFO:torch.nn.parallel.distributed:Reducer buckets have been rebuilt in this iteration." for a long time, I was trying to train on a 3 minutes speech, it took more than 5hrs for 20 epoch Thanks is advance