prevent sampling in first steps

2022-07-22 12:35:39 +00:00 · 2022-07-22 12:35:39 +00:00 · 9be5028617
parent 6dfe59e9b0
commit 9be5028617
1 changed files with 1 additions and 0 deletions
--- 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)