GPU memory about mp-senet HOT 5 CLOSED

1. I typically use two GPUs with 24 GB of memory each for training.
2. If you want to change the amount of dataset, you just need to edit the number of data indexes in MP-SENet/VoiceBank+DEMAND/training.txt and test.txt.

from mp-senet.

Dear Author,

Thank you for responding to my question. I'm not sure if the GPU memory used for this training can be reduced ...
Can I try to change the dense_channel to achieve this goal, or change some parameters (except batch_size)?

from mp-senet.

1. Sure, you can reduce GPU memory usage by decreasing the number of dense channels or by reducing the segment size in the config.json. However, these changes may affect the final training results.
2. By the way, you need to determine whether your GPU memory is insufficient during training or validation. During validation, the audio sequences are not truncated, which can cause variations in GPU memory usage. If you are experiencing memory issues only during validation, you can implement truncation as a solution.

from mp-senet.

Dear Author,

I try to decrease the number of dense channels in config.json to make it train successfully. However, when I implement the inference, I decreasing the number of dense channels in best_ckpt/config.json , it would appear size mismatch problem, I want to know the config.json in best_ckpt cannot change the dense channel ?

Sorry to bother the author's time !

from mp-senet.

The checkpoint file in best_ckpt is the pre-trained best checkpoint with the default dense channel, which is unchangeable. You should use your self-trained checkpoint for inference.

from mp-senet.