search

TreB1eN / InsightFace_Pytorch

Pytorch0.4.1 codes for InsightFace
MIT License
1.74k stars 423 forks source link

how to train on multiple GPUs? #130

Open ghost opened 4 years ago

ghost commented 4 years ago

I add
self.model = torch.nn.DataParallel(self.model).to(conf.device) self.head = torch.nn.DataParallel(self.head).to(conf.device)

but it has an error: 'DataParallel' object has no attribute 'kernel'. could you help me?