search

junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch
Other
23.09k stars 6.32k forks source link

Speed Up Training #1672

Open wyd2 opened 2 months ago

wyd2 commented 2 months ago

I used CycleGAN for CBCT-to-CT reconstruction. But the pace of this training is very slow. One eopch can take up to 6 hours. Is there any way to speed up the training？

kovalepskaia commented 2 months ago

you setting --batch_size and --num_threads