diff --git a/train_network.py b/train_network.py
index 07bf44bb..0fd8cc0f 100644
--- a/train_network.py
+++ b/train_network.py
@@ -365,6 +365,7 @@ def train(args):
         "ss_caption_tag_dropout_rate": args.caption_tag_dropout_rate,
         "ss_face_crop_aug_range": args.face_crop_aug_range,
         "ss_prior_loss_weight": args.prior_loss_weight,
+        "ss_min_snr_gamma": args.min_snr_gamma,
     }
 
     if use_user_config: