search

hanzhanggit / StackGAN-v2

MIT License
846 stars 190 forks source link

Can't support Multi GPU trainning #27

Closed jszgz closed 5 years ago

jszgz commented 5 years ago

torch.cuda.set_device(self.gpus[0])