unsupervised/config.yaml

training:
  num_epochs: 5000
  batch_size: 32
  adaptive_dataloader: false
  n_workers: 16
  optimizer:
    name: adamw
    lr: 0.0001
    weight_decay: 0
    encoder:
      weight_decay: 1.0e-06
  log:
    milestone: 20
    save:
      every: 1
      best: true
    train:
      reconstruction:
        every: 1
      images:
        every: 1
        how_many: 20
    val:
      reconstruction:
        every: 1
      error_rate:
        every: 0
        last: true
        kargs:
          dataloader_batch_size: 4
        eval_best: false
  device: 0
  seed: 97312
  flush_mem: true
  flush_per: 1
model:
  encoder:
    H: 96
    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.9529411764705882
      - 0.9019607843137255
      - 0.7529411764705882
  sprites:
    'n': 120
    color_channels: ${model.encoder.C}
    size:
    - ${eval:${model.encoder.H}//2}
    - ${eval:${model.encoder.H}//2}
    use_clamp: soft
    gen_type: unet
    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.01
dataset:
  copiale:
    path: copiale/
    sep: ' '
    space: _
    supervised: false
    crop_width: ${eval:${model.encoder.H}*2}
tag: a
timestamp: 2023-01-10_06-29-19
run_dir: /home/ysig/learnable-typewriter-supervised/runs/copiale/a/2023-01-10_06-29-19