From a7d302e196fa566809a169077a1a6412df42ed9b Mon Sep 17 00:00:00 2001 From: ddPn08 Date: Thu, 30 Mar 2023 16:49:03 +0900 Subject: [PATCH] write a random seed to metadata --- train_network.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/train_network.py b/train_network.py index b641e65c..c951b150 100644 --- a/train_network.py +++ b/train_network.py @@ -57,8 +57,9 @@ def train(args): use_dreambooth_method = args.in_json is None use_user_config = args.dataset_config is not None - if args.seed is not None: - set_seed(args.seed) + if args.seed is None: + args.seed = random.randint(0, 2**32) + set_seed(args.seed) tokenizer = train_util.load_tokenizer(args)