Fix Seed

2022-12-02 01:41:22 +09:00 · 2022-12-02 01:41:22 +09:00 · d225cd57e7
parent b96eabfe2a
commit d225cd57e7
1 changed files with 1 additions and 0 deletions
--- a/trainer/diffusers_trainer.py
+++ b/trainer/diffusers_trainer.py
@ -717,6 +717,7 @@ def main():

    # Set seed
    torch.manual_seed(args.seed)
+    random.seed(args.seed)
    print('RANDOM SEED:', args.seed)

    if args.resume: