training with multiple GPU

[gokceay]

Hello Mrs. Liyues,

I would like to use more than one GPU, how can I achieve this? In trainer net at line 25: self.model = nn.DataParallel(self.model).cuda(), I should add gpu ids inside DataParallel besides self.model right? Should I also increase the batch size to be more than the used GPU? Also did you tried batch size more than 1? What was the result if you tried? Thanks in advance.

[yinyin-llll]

hi，i also want to retrain the model but it need the csv file with annotation.Do you know this file or how to design this file?Thank you very much