From 3699a9064582c7309ab2578d84b28fc4b15deedc Mon Sep 17 00:00:00 2001 From: Kohya S Date: Mon, 15 May 2023 23:18:16 +0900 Subject: [PATCH] add adaptive noise scale to metadata --- train_network.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_network.py b/train_network.py index b331e92c..f2fd2009 100644 --- a/train_network.py +++ b/train_network.py @@ -347,6 +347,7 @@ def train(args): "ss_noise_offset": args.noise_offset, "ss_multires_noise_iterations": args.multires_noise_iterations, "ss_multires_noise_discount": args.multires_noise_discount, + "ss_adaptive_noise_scale": args.adaptive_noise_scale, "ss_training_comment": args.training_comment, # will not be updated after training "ss_sd_scripts_commit_hash": train_util.get_git_revision_hash(), "ss_optimizer": optimizer_name + (f"({optimizer_args})" if len(optimizer_args) > 0 else ""),