re-anchor, pretty meh results in-batch

10 months ago · 248d1a72f9
1 changed files with 2 additions and 2 deletions
--- 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