move the same content to hander.py

configs/synthesizers/UGATIT.yml

@@ -1,7 +1,7 @@
-name: selfie2anime
+name: selfie2anime-origin
 engine: UGATIT
 result_dir: ./result
-max_iteration: 100000
+max_pairs: 1000000
 
 distributed:
   model:
@@ -10,8 +10,15 @@ distributed:
 misc:
   random_seed: 324
 
-checkpoints:
-  interval: 1000
+checkpoint:
+  epoch_interval: 1 # one checkpoint every 1 epoch
+  n_saved: 5
+
+interval:
+  print_per_iteration: 10 # print once per 10 iteration
+  tensorboard:
+    scalar: 10
+    image: 1000
 
 model:
   generator:
@@ -26,12 +33,12 @@ model:
     _type: UGATIT-Discriminator
     in_channels: 3
     base_channels: 64
-    num_blocks: 3
+    num_blocks: 5
   global_discriminator:
     _type: UGATIT-Discriminator
     in_channels: 3
     base_channels: 64
-    num_blocks: 5
+    num_blocks: 7
 
 loss:
   gan:
@@ -62,9 +69,12 @@ optimizers:
 
 data:
   train:
+    scheduler:
+      start_proportion: 0.5
+      target_lr: 0
     buffer_size: 50
     dataloader:
-      batch_size: 8
+      batch_size: 4
       shuffle: True
       num_workers: 2
       pin_memory: True
@@ -85,9 +95,6 @@ data:
         - Normalize:
             mean: [0.5, 0.5, 0.5]
             std: [0.5, 0.5, 0.5]
-    scheduler:
-      start: 50000
-      target_lr: 0
   test:
     dataloader:
       batch_size: 4