From 80aca1ccc72679e658fb24d2fb7ea1998c1f240a Mon Sep 17 00:00:00 2001
From: rockerBOO <rockerboo@gmail.com>
Date: Tue, 5 Sep 2023 15:20:15 -0400
Subject: [PATCH] Add ip_noise_gamma metadata

---
 train_network.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train_network.py b/train_network.py
index f752607e..0f294675 100644
--- a/train_network.py
+++ b/train_network.py
@@ -509,6 +509,7 @@ class NetworkTrainer:
             "ss_prior_loss_weight": args.prior_loss_weight,
             "ss_min_snr_gamma": args.min_snr_gamma,
             "ss_scale_weight_norms": args.scale_weight_norms,
+            "ss_ip_noise_gamma": args.ip_noise_gamma,
         }
 
         if use_user_config: