add attention image fuse

2020-08-22 20:21:11 +08:00
parent ccc3d7614a
commit 58ed4524bf
4 changed files with 54 additions and 6 deletions
--- a/configs/synthesizers/UGATIT.yml
+++ b/configs/synthesizers/UGATIT.yml
@@ -12,13 +12,13 @@ misc:

 checkpoint:
  epoch_interval: 1 # one checkpoint every 1 epoch
-  n_saved: 5
+  n_saved: 2

 interval:
  print_per_iteration: 10 # print once per 10 iteration
  tensorboard:
    scalar: 10
-    image: 1000
+    image: 500

 model:
  generator: