diff options
Diffstat (limited to 'training/callbacks')
| -rw-r--r-- | training/callbacks/wandb_callbacks.py | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/training/callbacks/wandb_callbacks.py b/training/callbacks/wandb_callbacks.py index d9d81f6..451b0d5 100644 --- a/training/callbacks/wandb_callbacks.py +++ b/training/callbacks/wandb_callbacks.py @@ -178,13 +178,9 @@ class LogReconstuctedImages(Callback): experiment.log( { f"Reconstructions/{experiment.name}/{stage}": [ - [ - wandb.Image(img), - wandb.Image(rec), - ] + [wandb.Image(img), wandb.Image(rec),] for img, rec in zip( - imgs[: self.num_samples], - reconstructions[: self.num_samples], + imgs[: self.num_samples], reconstructions[: self.num_samples], ) ] } |