almost 0.1

configs/synthesizers/CyCleGAN.yml

@@ -1,40 +1,34 @@
-name: horse2zebra
-engine: cyclegan
+name: horse2zebra-CyCleGAN
+engine: CyCleGAN
 result_dir: ./result
-max_iteration: 16600
-
-distributed:
-  model:
-    # broadcast_buffers: False
+max_pairs: 266800
 
 misc:
   random_seed: 324
 
-checkpoints:
-  interval: 2000
-
-log:
-  logger:
-    level: 20 # DEBUG(10) INFO(20)
+handler:
+  clear_cuda_cache: False
+  set_epoch_for_dist_sampler: True
+  checkpoint:
+    epoch_interval: 1 # checkpoint once per `epoch_interval` epoch
+    n_saved: 2
+  tensorboard:
+    scalar: 100 # log scalar `scalar` times per epoch
+    image: 2 # log image `image` times per epoch
 
 model:
   generator:
-    _type: ResGenerator
+    _type: CyCle-Generator
     in_channels: 3
     out_channels: 3
     base_channels: 64
     num_blocks: 9
     padding_mode: reflect
     norm_type: IN
-    use_dropout: False
   discriminator:
     _type: PatchDiscriminator
-#    _distributed:
-#      bn_to_syncbn: False
     in_channels: 3
     base_channels: 64
-    num_conv: 3
-    norm_type: IN
 
 loss:
   gan:
@@ -53,19 +47,22 @@ optimizers:
   generator:
     _type: Adam
     lr: 2e-4
-    betas: [0.5, 0.999]
+    betas: [ 0.5, 0.999 ]
   discriminator:
     _type: Adam
     lr: 2e-4
-    betas: [0.5, 0.999]
+    betas: [ 0.5, 0.999 ]
 
 data:
   train:
+    scheduler:
+      start_proportion: 0.5
+      target_lr: 0
     buffer_size: 50
     dataloader:
-      batch_size: 16
+      batch_size: 6
       shuffle: True
-      num_workers: 4
+      num_workers: 2
       pin_memory: True
       drop_last: True
     dataset:
@@ -76,14 +73,14 @@ data:
       pipeline:
         - Load
         - Resize:
-            size: [286, 286]
+            size: [ 286, 286 ]
         - RandomCrop:
-            size: [256, 256]
+            size: [ 256, 256 ]
         - RandomHorizontalFlip
         - ToTensor
-    scheduler:
-      start: 8300
-      target_lr: 0
+        - Normalize:
+            mean: [ 0.5, 0.5, 0.5 ]
+            std: [ 0.5, 0.5, 0.5 ]
   test:
     dataloader:
       batch_size: 4
@@ -99,5 +96,8 @@ data:
       pipeline:
         - Load
         - Resize:
-            size: [256, 256]
+            size: [ 256, 256 ]
         - ToTensor
+        - Normalize:
+            mean: [ 0.5, 0.5, 0.5 ]
+            std: [ 0.5, 0.5, 0.5 ]

configs/synthesizers/TAFG.yml

@@ -1,7 +1,7 @@
 name: TAFG
 engine: TAFG
 result_dir: ./result
-max_pairs: 1000000
+max_pairs: 1500000
 
 handler:
   clear_cuda_cache: True
@@ -28,7 +28,7 @@ model:
     _type: MultiScaleDiscriminator
     num_scale: 2
     discriminator_cfg:
-      _type: pix2pixHD-PatchDiscriminator
+      _type: PatchDiscriminator
       in_channels: 3
       base_channels: 64
       use_spectral: True