summaryrefslogtreecommitdiff
path: root/training/callbacks
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-28 15:14:55 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-28 15:14:55 +0200
commitc032ffb05a7ed86f8fe5d596f94e8997c558cae8 (patch)
treebf890ffd4c815db7d510cfb281d253b5728f70c6 /training/callbacks
parent524bf4351ac295bd4ff9914bb1f32eda7f7ff855 (diff)
Reformatting with attrs, config for encoder and decoder
Diffstat (limited to 'training/callbacks')
-rw-r--r--training/callbacks/wandb_callbacks.py8
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],
)
]
}