supervised/config.yaml

training:
  num_epochs: 100
  batch_size: 16
  adaptive_dataloader: false
  n_workers: 8
  optimizer:
    name: adamw
    lr: 0.001
    weight_decay: 0
    encoder:
      weight_decay: 1.0e-06
  log:
    milestone: 1
    save:
      every: 1
      best: true
    train:
      reconstruction:
        every: 1
      images:
        every: 1
        how_many: 20
    val:
      reconstruction:
        every: 1
      error_rate:
        every: 1
        last: true
        kargs:
          dataloader_batch_size: 4
  device: 0
  seed: 97312
  flush_mem: true
  flush_per: 1
model:
  encoder:
    H: 64
    C: 3
    name: resnet32
    pooling_on_width: 4
  background:
    size:
    - ${eval:${model.encoder.H}//2}
    - ${model.encoder.H}
    use_clamp: soft
    init:
      freeze: true
      constant:
      - 0.7411764705882353
      - 0.5568627450980392
      - 0.3764705882352941
  sprites:
    'n': 83
    color_channels: ${model.encoder.C}
    size:
    - ${eval:${model.encoder.H}//2}
    - ${eval:${model.encoder.H}//2}
    use_clamp: soft
    gen_type: mlp
    init:
      color:
        freeze: true
        constant: 0.5
  transformation:
    canvas_size:
    - ${model.encoder.H}
    - ${eval:${model.encoder.H}*2}
    layer:
      ops:
      - color
      - position
      color:
        color_channels: ${model.encoder.C}
        n_hidden_layers: 2
        n_hidden_units: 128
      position:
        n_hidden_layers: 2
        n_hidden_units: 128
        max_x: 1.0
        max_y: 1.0
        parametrization: exp
        padding_mode: zeros
    background:
      ops:
      - color
      color:
        color_channels: ${model.encoder.C}
        n_hidden_layers: 2
        n_hidden_units: 128
  window:
    size: ${model.encoder.H}
  loss:
    sigma_gaussian: 7
    ctc_factor: 0.1
dataset:
  google:
    path: google
    supervised: true
tag: supervised
timestamp: 2023-01-06_21-14-43
run_dir: /home/ysig/learnable-typewriter-supervised/runs/google/supervised/2023-01-06_21-14-43