TAFG good result

configs/synthesizers/TAFG.yml

@@ -23,7 +23,8 @@ model:
     _bn_to_sync_bn: False
     style_in_channels: 3
     content_in_channels: 24
-    num_blocks: 8
+    num_adain_blocks: 8
+    num_res_blocks: 0
   discriminator:
     _type: MultiScaleDiscriminator
     num_scale: 2
@@ -47,21 +48,17 @@ loss:
       "11": 0.125
       "20": 0.25
       "29": 1
-    criterion: 'L2'
+    criterion: 'L1'
     style_loss: False
     perceptual_loss: True
-    weight: 0.5
+    weight: 10
   style:
     layer_weights:
-      "1": 0.03125
-      "6": 0.0625
-      "11": 0.125
-      "20": 0.25
-      "29": 1
-    criterion: 'L2'
+      "3": 1
+    criterion: 'L1'
     style_loss: True
     perceptual_loss: False
-    weight: 0
+    weight: 10
   fm:
     level: 1
     weight: 10
@@ -71,6 +68,9 @@ loss:
   style_recon:
     level: 1
     weight: 0
+  edge:
+    weight: 10
+    hed_pretrained_model_path: ./network-bsds500.pytorch
 
 optimizers:
   generator:
@@ -91,7 +91,7 @@ data:
       target_lr: 0
     buffer_size: 50
     dataloader:
-      batch_size: 24
+      batch_size: 8
       shuffle: True
       num_workers: 2
       pin_memory: True