Closer

jloveric · May 20, 2024 · 462747e · 462747e
1 parent f16c8e7
commit 462747e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):