From 1457145d46e353688ce6e5ff4d13577d87971e3c Mon Sep 17 00:00:00 2001 From: Nathan Date: Wed, 20 Nov 2024 10:29:50 -0600 Subject: [PATCH] Remove unsupported 'range' argument in make_grid --- train_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_vae.py b/train_vae.py index 7a1e5b3c..6590d12a 100644 --- a/train_vae.py +++ b/train_vae.py @@ -259,7 +259,7 @@ def save_model(path): images, recons = map(lambda t: t[:k], (images, recons)) images, recons, hard_recons, codes = map(lambda t: t.detach().cpu(), (images, recons, hard_recons, codes)) - images, recons, hard_recons = map(lambda t: make_grid(t.float(), nrow = int(sqrt(k)), normalize = True, range = (-1, 1)), (images, recons, hard_recons)) + images, recons, hard_recons = map(lambda t: make_grid(t.float(), nrow = int(sqrt(k)), normalize = True), (images, recons, hard_recons)) logs = { **logs,