config/model/painter/opts.yaml

comet:
  display_size: 25
  im_per_row:
    d: 3
    m: 4
    p: 4
    s: 3
  rows_per_log: 5
  tags: []

data:
  check_samples: false
  files:
    base: /network/tmp1/ccai/data/omnigan/twitter/
    train:
      r: /network/tmp1/ccai/data/omnigan/twitter/train_r.json
      rf: /network/tmp1/ccai/data/omnigan/twitter/train_rf.json
      s: /network/tmp1/ccai/data/omnigan/twitter/train_s.json
    val:
      r: /network/tmp1/ccai/data/omnigan/twitter/val_r.json
      rf: /network/tmp1/ccai/data/omnigan/twitter/val_rf_labelbox.json
      s: /network/tmp1/ccai/data/omnigan/twitter/val_s.json
  loaders:
    batch_size: 6
    num_workers: 6
    shuffle: false
  max_samples: -1
  normalization: default
  transforms:
  - ignore: false
    name: hflip
    p: 0.5
  - ignore: false
    keep_aspect_ratio: true
    name: resize
    new_size: 640
  - height: 600
    ignore: false
    name: crop
    width: 600
  - ignore: false
    name: resize
    new_size: 640
dis:
  default:
    get_intermediate_features: false
    init_gain: 0.2
    init_type: kaiming
    input_nc: 3
    n_layers: 4
    ndf: 64
    norm: instance
    num_D: 1
    use_sigmoid: false
  flip_prob: 0.05
  m:
    architecture: base
    gan_type: WGAN_norm
    get_intermediate_features: false
    init_gain: 0.2
    init_type: kaiming
    input_nc: 3
    multi_level: false
    n_layers: 4
    ndf: 64
    norm: instance
    num_D: 1
    use_sigmoid: false
    wgan_clamp_lower: -0.01
    wgan_clamp_upper: 0.01
  opt:
    beta1: 0.5
    lr:
      default: 0.001
    lr_gamma: 0.5
    lr_milestones: 15
    lr_policy: step
    lr_step_size: 30
    optimizer: ExtraAdam
  p:
    get_intermediate_features: true
    init_gain: 0.2
    init_type: kaiming
    input_nc: 3
    n_layers: 4
    ndf: 64
    norm: batch
    num_D: 3
    use_local_discriminator: false
    use_sigmoid: false
  s:
    gan_type: WGAN_norm
    get_intermediate_features: false
    init_gain: 0.2
    init_type: kaiming
    input_nc: 3
    n_layers: 4
    ndf: 64
    norm: instance
    num_D: 1
    use_sigmoid: false
    wgan_clamp_lower: -0.01
    wgan_clamp_upper: 0.01
  soft_shift: 0.2
domains:
- rf
gen:
  d:
    activ: lrelu
    conv_norm: batch
    init_gain: 0.01
    init_type: normal
    n_downsample: 3
    n_res: 1
    n_upsample: 3
    output_dim: 1
    pad_type: reflect
    proj_dim: 32
    res_dim: 2048
    res_norm: batch
  deeplabv2:
    nblocks:
    - 3
    - 4
    - 23
    - 3
    pretrained_model: /network/tmp1/ccai/data/omnigan/pretrained_models/DeepLab_resnet_pretrained_imagenet.pth
    use_pretrained: true
  default:
    activ: lrelu
    conv_norm: spectral
    init_gain: 0.2
    init_type: xavier
    n_downsample: 3
    n_res: 1
    n_upsample: 3
    pad_type: reflect
    proj_dim: 32
    res_dim: 2048
    res_norm: spectral
  encoder:
    activ: lrelu
    architecture: deeplabv2
    conv_norm: spectral
    dim: 32
    init_gain: 0.2
    init_type: xavier
    input_dim: 3
    n_downsample: 3
    n_res: 0
    n_upsample: 3
    norm: spectral
    pad_type: reflect
    proj_dim: 32
    res_dim: 2048
    res_norm: spectral
  m:
    activ: lrelu
    conv_norm: spectral
    init_gain: 0.2
    init_type: xavier
    loss: gan
    n_downsample: 3
    n_res: 1
    n_upsample: 3
    output_dim: 1
    pad_type: reflect
    proj_dim: 32
    res_dim: 2048
    res_norm: spectral
    use_advent: true
    use_ground_intersection: true
    use_minent: true
    use_minent_var: true
  opt:
    beta1: 0.5
    lr:
      default: 0.0005
    lr_gamma: 0.5
    lr_milestones: 15
    lr_policy: step
    lr_step_size: 30
    optimizer: ExtraAdam
  p:
    activ: relu
    conv_norm: spectral
    init_gain: 0.2
    init_type: kaiming
    latent_dim: 640
    loss: gan
    n_downsample: 3
    n_interm_res: 11
    n_res: 1
    n_upsample: 3
    no_z: true
    output_dim: 3
    pad_type: reflect
    paste_original_content: true
    proj_dim: 32
    res_dim: 2048
    res_norm: spectral
    spade_kernel_size: 3
    spade_n_up: 7
    spade_param_free_norm: instance
    spade_use_spectral_norm: true
    use_final_shortcut: false
  s:
    activ: lrelu
    conv_norm: spectral
    init_gain: 0.2
    init_type: xavier
    n_downsample: 3
    n_res: 1
    n_upsample: 3
    num_classes: 11
    output_dim: 11
    pad_type: reflect
    proj_dim: 32
    res_dim: 2048
    res_norm: spectral
    upsample_featuremaps: false
    use_advent: true
    use_minent: true
    use_pseudo_labels: false
load_paths:
  m: none
  p: none
output_path: /miniscratch/schmidtv/vicc/omnigan/runs/painter/default (2)
tasks:
- p
train:
  auto_resume: true
  epochs: 10000
  fid:
    batch_size: 50
    dims: 2048
    n_images: 57
  lambdas:
    C: 1
    G:
      classifier: 1
      d:
        gml: 0.5
        main: 1
      m:
        bce: 1
        gi: 0.05
        tv: 1
      p:
        context: 0
        dm: 1
        featmatch: 10
        gan: 1
        reconstruction: 0
        tv: 0
        vgg: 10
      s:
        advent: 0.001
        crossent: 1
        crossent_pseudo: 0.001
        minent: 0.001
    advent:
      WGAN_gp: 10
      adv_aux: 0.0
      adv_main: 1.0
      dis_aux: 0.0
      dis_main: 1.0
      ent_aux: 0.0
      ent_main: 0.5
      ent_var: 0.1
  latent_domain_adaptation: false
  log_level: 2
  resume: false
  save_n_epochs: 1
val:
  store_images: false