search

open-mmlab / Amphion

Amphion (/æmˈfaɪən/) is a toolkit for Audio, Music, and Speech Generation. Its purpose is to support reproducible research and help junior researchers and engineers get started in the field of audio, music, and speech generation research and development.
https://openhlt.github.io/amphion/
MIT License
4.41k stars 373 forks source link

[Help]: Need a list of hardware configurations. #151

Closed isKEKE closed 5 months ago

isKEKE commented 5 months ago

Hello Team

What is the minimum GPU memory to train TTA? I tried to train with 4060TI but it says not enough memory.

Is it possible to provide a detailed list of hardware configuration?

Thanks

HeCheng0625 commented 5 months ago

Hi, we use one 24G 4090 GPU to train the TTA model, you can decrease the batch size (you may need decrease the learning rate at the same time) if you don't have enough RAM.

isKEKE commented 5 months ago

Thanks for the advice.