New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

What is the requirement in terms of hardware? #19

Closed

underdogliu opened this issue Jul 26, 2021 · 1 comment

underdogliu commented Jul 26, 2021

First of all thanks for such amazing work of RawNet and its variants!

I just want to know about the hardware requirements for training RawNet2 and modified RawNet2. I have following two questions:

I tried to run it with a single GPU with 12GB memory and failed. What was your experimental hardware setting?
Is it recommended to reduce number of mini batches for better memory handling?

Owner

Jungjee commented Jul 29, 2021

Hi,

I recall using 4 Nvidia Titan V GPUs.
You can reduce the size of mini-batch, however, it would take a lot of time to train VoxCeleb2 using a single GPU
(regardless of your GPU generation, normally many recent systems take days using 4 GPUs )

My quick advice is to train the model using the train data of VoxCeleb1 and evaluate using VoxCeleb1 testset.

Jungjee closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment