From bc82188377a8dc3068cd7b2ba5f02ed8bd4f2a54 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Wed, 7 Apr 2021 15:46:34 +0800
Subject: Revert cross-reconstruction loss factor and make image log steps
 adjustable

---
 config.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'config.py')

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,
     },
-- 
cgit v1.2.3