From 9be50286174ca3b9f94808357cf031901543eb88 Mon Sep 17 00:00:00 2001
From: Patrick Esser <patrick.esser@gmx.net>
Date: Fri, 22 Jul 2022 12:35:39 +0000
Subject: [PATCH] prevent sampling in first steps

---
 main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/main.py b/main.py
index b190065..21e8135 100644
--- a/main.py
+++ b/main.py
@@ -350,6 +350,7 @@ class ImageLogger(Callback):
         if (self.check_frequency(check_idx) and  # batch_idx % self.batch_freq == 0
                 hasattr(pl_module, "log_images") and
                 callable(pl_module.log_images) and
+                batch_idx > 5 and
                 self.max_images > 0):
             logger = type(pl_module.logger)