search

HsinYingLee / DRIT

Learning diverse image-to-image translation from unpaired data
844 stars 155 forks source link

How can I train with multi-gpus? #49

Open haezu opened 4 years ago

haezu commented 4 years ago

How can I train with multi-gpus? nn.Dataparallel does't work.

hytseng0509 commented 4 years ago

You can refer to the CycleGAN implementation for multi-gpus training.

KevinMarkVine commented 3 years ago

How can I train with multi-gpus? nn.Dataparallel does't work.

Did you solve this problem？