Upload 4 files

supervised/config.yaml

@@ -0,0 +1,101 @@
+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

supervised/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a4076300fd188075f3d7e74188e85b0c34252afe9e534465315ca7bc5d0a1f65
+size 47392127

unsupervised/config.yaml

@@ -0,0 +1,103 @@
+training:
+  num_epochs: 1000
+  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: 1
+    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: 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': 60
+    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:
+  google:
+    path: google
+    supervised: false
+    crop_width: 128
+tag: unsupervised
+timestamp: 2023-01-12_02-53-55
+run_dir: /home/ysig/learnable-typewriter-supervised/runs/google/unsupervised/2023-01-12_02-53-55

unsupervised/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c639003bfaee19c33a5b360a8c5d97bdd591b257ebae88d3efd76dcdbbfa0f1a
+size 10741767