printing training data's pixel values with """print(f"{batch['pixel_values'].shape=}")"""

train_dreambooth_lora_sdxl_advanced.py

@@ -1893,6 +1893,7 @@ def main(args):
         latents_cache = []
         for batch in tqdm(train_dataloader, desc="Caching latents"):
             with torch.no_grad():
+                print(f"{batch['pixel_values'].shape=}")
                 batch["pixel_values"] = batch["pixel_values"].to(
                     accelerator.device, non_blocking=True, dtype=torch.float32
                 )