TAHG 0.0.3

configs/synthesizers/TAHG.yml

@@ -17,14 +17,15 @@ checkpoint:
 interval:
   print_per_iteration: 10 # print once per 10 iteration
   tensorboard:
-    scalar: 10
-    image: 500
+    scalar: 100
+    image: 2
 
 model:
   generator:
     _type: TAHG-Generator
     style_in_channels: 3
     content_in_channels: 1
+    num_blocks: 4
   discriminator:
     _type: TAHG-Discriminator
     in_channels: 3
@@ -37,22 +38,22 @@ loss:
     weight: 1.0
   edge:
     criterion: 'L1'
-    hed_pretrained_model_path: "/root/network-bsds500.pytorch"
-    weight: 2
+    hed_pretrained_model_path: "./network-bsds500.pytorch"
+    weight: 1
   perceptual:
     layer_weights:
-#      "3": 1.0
-      "0": 1.0
-      "5": 1.0
-      "10": 1.0
-      "19": 1.0
+      "3": 1.0
+#      "0": 1.0
+#      "5": 1.0
+#      "10": 1.0
+#      "19": 1.0
     criterion: 'L2'
     style_loss: True
     perceptual_loss: False
-    weight: 100.0
+    weight: 20
   recon:
     level: 1
-    weight: 2
+    weight: 1
 
 optimizers:
   generator:
@@ -73,7 +74,7 @@ data:
       target_lr: 0
     buffer_size: 50
     dataloader:
-      batch_size: 48
+      batch_size: 160
       shuffle: True
       num_workers: 2
       pin_memory: True