rewrite

configs/synthesizers/UGATIT.yml

@@ -14,11 +14,15 @@ handler:
     n_saved: 2
   tensorboard:
     scalar: 100 # log scalar `scalar` times per epoch
-    image: 2 # log image `image` times per epoch
+    image: 4 # log image `image` times per epoch
+  test:
+    random: True
+    images: 10
 
 model:
   generator:
     _type: UGATIT-Generator
+    _add_spectral_norm: True
     in_channels: 3
     out_channels: 3
     base_channels: 64
@@ -27,11 +31,13 @@ model:
     light: True
   local_discriminator:
     _type: UGATIT-Discriminator
+    _add_spectral_norm: True
     in_channels: 3
     base_channels: 64
     num_blocks: 5
   global_discriminator:
     _type: UGATIT-Discriminator
+    _add_spectral_norm: True
     in_channels: 3
     base_channels: 64
     num_blocks: 7
@@ -50,6 +56,8 @@ loss:
     weight: 10.0
   cam:
     weight: 1000
+  mgc:
+    weight: 0
 
 optimizers:
   generator:
@@ -70,7 +78,7 @@ data:
       target_lr: 0
     buffer_size: 50
     dataloader:
-      batch_size: 24
+      batch_size: 4
       shuffle: True
       num_workers: 2
       pin_memory: True