aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2025-12-20 14:54:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixing validation step for PL 1

Browse Source
This commit is contained in:
Tim
2023-06-30 13:29:02 -07:00
parent 2796c81a5f
commit 061d11d55d
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@@ -469,9 +469,8 @@ class ImageLogger(Callback):
self.log_img(pl_module, batch, batch_idx, split="train")
@rank_zero_only
# def on_validation_batch_end(self, trainer, pl_module, outputs, batch, batch_idx):
def on_validation_batch_end(
self, trainer, pl_module, outputs, batch, batch_idx, **kwargs
self, trainer, pl_module, outputs, batch, batch_idx, *args, **kwargs
):
if not self.disabled and pl_module.global_step > 0:
self.log_img(pl_module, batch, batch_idx, split="val")