From 263bc7e04915d68fb669d4f77caa8f8d8730f847 Mon Sep 17 00:00:00 2001 From: Luke Wood Date: Sun, 13 Feb 2022 21:51:59 -0800 Subject: [PATCH] Tiny readability update --- projects/4-fashion-autoencoder/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/4-fashion-autoencoder/train.py b/projects/4-fashion-autoencoder/train.py index 46223aae3..3f2d95b10 100644 --- a/projects/4-fashion-autoencoder/train.py +++ b/projects/4-fashion-autoencoder/train.py @@ -42,8 +42,8 @@ def log_images(epoch, logs): pred_data = np.clip(model.predict(test_data), 0, 1) wandb.log({ "examples": [ - wandb.Image(np.hstack([data, pred_data[i]]), caption=str(i)) - for i, data in enumerate(test_data)] + wandb.Image(np.hstack([data, pred]), caption=str(i)) + for i, (pred, data) in zip(test_data, pred_data)] }, commit=False)