TAFG update

configs/synthesizers/TAFG.yml

@@ -1,7 +1,7 @@
 name: TAFG-vox2
 engine: TAFG
 result_dir: ./result
-max_pairs: 1500000
+max_pairs: 1000000
 
 handler:
   clear_cuda_cache: True
@@ -12,10 +12,13 @@ handler:
   tensorboard:
     scalar: 100 # log scalar `scalar` times per epoch
     image: 4 # log image `image` times per epoch
+  test:
+    random: True
+    images: 10
 
 
 misc:
-  random_seed: 123
+  random_seed: 1004
 
 model:
   generator:
@@ -23,10 +26,13 @@ model:
     _bn_to_sync_bn: False
     style_in_channels: 3
     content_in_channels: 24
-    num_adain_blocks: 8
-    num_res_blocks: 8
-    use_spectral_norm: True
-    style_use_fc: False
+    use_spectral_norm: False
+    style_encoder_type: StyleEncoder
+    num_style_conv: 4
+    style_dim: 8
+    num_adain_blocks: 4
+    num_res_blocks: 4
+    
   discriminator:
     _type: MultiScaleDiscriminator
     num_scale: 2
@@ -54,17 +60,24 @@ loss:
     style_loss: False
     perceptual_loss: True
     weight: 0
+  style:
+    layer_weights:
+      "3": 1
+    criterion: 'L1'
+    style_loss: True
+    perceptual_loss: False
+    weight: 10
   recon:
     level: 1
     weight: 10
   style_recon:
     level: 1
-    weight: 5
+    weight: 1
   content_recon:
     level: 1
-    weight: 10
+    weight: 1
   edge:
-    weight: 10
+    weight: 5
     hed_pretrained_model_path: ./network-bsds500.pytorch
   cycle:
     level: 1
@@ -89,7 +102,7 @@ data:
       target_lr: 0
     buffer_size: 50
     dataloader:
-      batch_size: 1
+      batch_size: 8
       shuffle: True
       num_workers: 1
       pin_memory: True