From 248d1a72f94f7ff6de2b09dc135f41ee6aea3714 Mon Sep 17 00:00:00 2001
From: "Michael Pilosov, PhD" <consistentbayes@gmail.com>
Date: Sun, 28 Jan 2024 01:52:18 +0000
Subject: [PATCH] re-anchor, pretty meh results in-batch

---
 model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model.py b/model.py
index 5b1eb4c..4dce517 100644
--- a/model.py
+++ b/model.py
@@ -52,8 +52,8 @@ class ColorTransformerModel(L.LightningModule):
         p_loss = preservation_loss(
             inputs,
             outputs,
-            # target_inputs=rgb_tensor,
-            # target_outputs=self.forward(rgb_tensor),
+            target_inputs=rgb_tensor,
+            target_outputs=self.forward(rgb_tensor),
         )
         alpha = self.hparams.alpha