Added metadata

hiddenswitch · Jul 9, 2024 · 95da56b · 95da56b
1 parent 4e0af51
commit 95da56b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/sd_scripts/stable_cascade_train_c_network.py b/sd_scripts/stable_cascade_train_c_network.py
@@ -628,6 +628,9 @@ def train(self, args):
             "ss_scale_weight_norms": args.scale_weight_norms,
             "ss_ip_noise_gamma": args.ip_noise_gamma,
             "ss_debiased_estimation": bool(args.debiased_estimation_loss),
+            "ss_sigloss": bool(args.use_sig_loss),
+            "ss_dataset": args.train_data_dir,
+            "ss_stdweight": args.std_loss_weight,
         }
 
         if use_user_config:

diff --git a/sd_scripts/train_network.py b/sd_scripts/train_network.py
@@ -693,6 +693,9 @@ def train(self, args):
             "ss_scale_weight_norms": args.scale_weight_norms,
             "ss_ip_noise_gamma": args.ip_noise_gamma,
             "ss_debiased_estimation": bool(args.debiased_estimation_loss),
+            "ss_sigloss": bool(args.use_sig_loss),
+            "ss_dataset": args.train_data_dir,
+            "ss_stdweight": args.std_loss_weight,
         }
 
         if use_user_config: