VNext train IDOL question

train IDOL question

Open nazimii opened this issue 2 years ago • 1 comments

hi, I have two question: (1)when train IDOL, how many GPU you use? (2)In config, your batch size is 32,and it is too big for me,

Dec 23 '22 02:12 nazimii

In the paper, you say: The model is trained on 8 V100 GPUs of 32G RAM, with 2 pairs of frames per GPU. so, the batchsize is 4 at each GPU? 8*4 = 32?

Dec 23 '22 02:12 nazimii