Revert cross-reconstruction loss factor and make image log steps adjustable

1 files changed, 2 insertions, 0 deletions

diff --git a/config.py b/config.py
index 04f7b36..df7c64c 100644
--- a/config.py
+++ b/config.py
@@ -10,6 +10,8 @@ config: Configuration = {
         'save_dir': 'runs',
         # Recorde disentangled image or not
         'image_log_on': False,
+        # Image log per n steps
+        'image_log_steps': 100,
         # The number of subjects for validating (Part of testing set)
         'val_size': 20,
     },