ptrblck:

self.module1 = nn.DataParallel(self.module1, device_ids=[0, 1]).to(‘cuda:0’)


can you explain what does this means, so you are using gpus 0 and 1 to be used for datapralel and then what .to('cuda:0') means/does?