From 3ee13893f9a4973ac75f45fe9318c35760dd4b1f Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 7 Jan 2023 13:57:46 +0100 Subject: Added progressive aspect ratio bucketing --- training/util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'training/util.py') diff --git a/training/util.py b/training/util.py index bc466e2..6f42228 100644 --- a/training/util.py +++ b/training/util.py @@ -58,8 +58,8 @@ class CheckpointerBase: def save_samples(self, pipeline, step, num_inference_steps, guidance_scale=7.5, eta=0.0): samples_path = Path(self.output_dir).joinpath("samples") - train_data = self.datamodule.train_dataloader() - val_data = self.datamodule.val_dataloader() + train_data = self.datamodule.train_dataloaders[0] + val_data = self.datamodule.val_dataloader generator = torch.Generator(device=pipeline.device).manual_seed(self.seed) -- cgit v1.2.3-54-g00ecf