name: horse2zebra
engine: cyclegan
result_dir: ./result
max_iteration: 16600

distributed:
  model:
    # broadcast_buffers: False

misc:
  random_seed: 324

checkpoints:
  interval: 2000

log:
  logger:
    level: 20 # DEBUG(10) INFO(20)

model:
  generator:
    _type: ResGenerator
    in_channels: 3
    out_channels: 3
    base_channels: 64
    num_blocks: 9
    padding_mode: reflect
    norm_type: IN
    use_dropout: False
  discriminator:
    _type: PatchDiscriminator
#    _distributed:
#      bn_to_syncbn: False
    in_channels: 3
    base_channels: 64
    num_conv: 3
    norm_type: IN

loss:
  gan:
    loss_type: lsgan
    weight: 1.0
    real_label_val: 1.0
    fake_label_val: 0.0
  cycle:
    level: 1
    weight: 10.0
  id:
    level: 1
    weight: 0

optimizers:
  generator:
    _type: Adam
    lr: 2e-4
    betas: [0.5, 0.999]
  discriminator:
    _type: Adam
    lr: 2e-4
    betas: [0.5, 0.999]

data:
  train:
    buffer_size: 50
    dataloader:
      batch_size: 16
      shuffle: True
      num_workers: 4
      pin_memory: True
      drop_last: True
    dataset:
      _type: GenerationUnpairedDataset
      root_a: "/data/i2i/horse2zebra/trainA"
      root_b: "/data/i2i/horse2zebra/trainB"
      random_pair: True
      pipeline:
        - Load
        - Resize:
            size: [286, 286]
        - RandomCrop:
            size: [256, 256]
        - RandomHorizontalFlip
        - ToTensor
    scheduler:
      start: 8300
      target_lr: 0
  test:
    dataloader:
      batch_size: 4
      shuffle: False
      num_workers: 1
      pin_memory: False
      drop_last: False
    dataset:
      _type: GenerationUnpairedDataset
      root_a: "/data/i2i/horse2zebra/testA"
      root_b: "/data/i2i/horse2zebra/testB"
      random_pair: False
      pipeline:
        - Load
        - Resize:
            size: [256, 256]
        - ToTensor