The code is distributed training implementation for CycleGan.

Original paper: https://arxiv.org/abs/1703.10593

Official source code: https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

• Clone this repo:

git clone https://github.com/RyanWangCode/CycleGAN-horovod
cd pytorch-CycleGAN-and-pix2pix

• Install Horovod, PyTorch 0.4+ and other dependencies (e.g., torchvision).

• Download a CycleGAN dataset (e.g. maps):

bash ./datasets/download_cyclegan_dataset.sh maps

• To view training results and loss plots, run python -m visdom.server and click the URL http://localhost:8097.
• Train a model:

#!./scripts/train_cyclegan.sh
python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan

To see more intermediate results, check out ./checkpoints/maps_cyclegan/web/index.html.

• Test the model:

#!./scripts/test_cyclegan.sh
python test.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan

• The test results will be saved to a html file here: ./results/maps_cyclegan/latest_test/index.html.