Upload 4 files

supervised/config.yaml

@@ -0,0 +1,103 @@
+training:
+  num_epochs: 500
+  batch_size: 16
+  adaptive_dataloader: false
+  n_workers: 8
+  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: 1
+        last: true
+        kargs:
+          dataloader_batch_size: 4
+  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': 112
+    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.01
+dataset:
+  copiale:
+    path: copiale/
+    sep: ' '
+    space: _
+    supervised: true
+tag: supervised
+timestamp: 2023-01-05_02-25-55
+run_dir: /home/ysig/learnable-typewriter-supervised/runs/copiale/supervised/2023-01-05_02-25-55

supervised/model.pth

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

unsupervised/config.yaml

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

unsupervised/model.pth

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