From 462747ec1c9d2e74d8cdd89315258bf204aba893 Mon Sep 17 00:00:00 2001 From: jloveric Date: Sun, 19 May 2024 21:16:51 -0700 Subject: [PATCH] Closer --- high_order_implicit_representation/rendering.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/high_order_implicit_representation/rendering.py b/high_order_implicit_representation/rendering.py index 3c45346..5d9c22b 100644 --- a/high_order_implicit_representation/rendering.py +++ b/high_order_implicit_representation/rendering.py @@ -291,7 +291,7 @@ def on_train_epoch_end( f, axarr = plt.subplots(1, 2) axarr[0].imshow(ans.detach().cpu().numpy()) axarr[0].set_title("fit") - axarr[1].imshow(self.image.cpu()) + axarr[1].imshow(image.cpu()) axarr[1].set_title("original") for i in range(2):