From 8d14893598fd9dc400cc2144b27f0e7bd1f71106 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Fri, 7 Feb 2025 01:28:26 +0800 Subject: [PATCH] Update train_network.py --- train_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_network.py b/train_network.py index e7c67eb4..9f4e3a87 100644 --- a/train_network.py +++ b/train_network.py @@ -1086,7 +1086,7 @@ class NetworkTrainer: if args.save_state: train_util.save_and_remove_state_on_epoch_end(args, accelerator, epoch + 1) - if args.sample_every_n_epochs is not None and (epoch + 1)% args.sample_every_n_epochs != 0: + if args.sample_every_n_epochs is not None and (epoch + 1)% args.sample_every_n_epochs == 0: example_tuple = (latents, batch["captions"]) self.sample_images(accelerator, args, epoch + 1, global_step, accelerator.device, vae, tokenizer, text_encoder, unet, example_tuple)