RVC-Project/Retrieval-based-Voice-Conversion-WebUI
MacOS | MPS | torch metal work on Train -> step-1 then Step-2 but not on training the model (Step-3)
オープン
#991 opened on 2023/08/12
help wanted
Repository metrics
- Stars
- (18,427 個のスター)
- PR merge metrics
- (30d に merged 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