From 626d4b433a64b030cfe67df547467939b2b572c3 Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Mon, 3 Apr 2023 12:38:20 -0400 Subject: [PATCH] Add min_snr_gamma to metadata --- train_network.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_network.py b/train_network.py index 476f76df..da91ec4f 100644 --- a/train_network.py +++ b/train_network.py @@ -346,6 +346,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: