diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt
index 3f966d3..ea5e856 100644
--- a/checkpoints/ROMA_UNSB_001/loss_log.txt
+++ b/checkpoints/ROMA_UNSB_001/loss_log.txt
@@ -2,3 +2,4 @@
 ================ Training Loss (Sun Feb 23 15:52:29 2025) ================
 ================ Training Loss (Sun Feb 23 16:00:07 2025) ================
 ================ Training Loss (Sun Feb 23 16:02:40 2025) ================
+================ Training Loss (Sun Feb 23 16:05:19 2025) ================
diff --git a/models/__pycache__/roma_unsb_model.cpython-39.pyc b/models/__pycache__/roma_unsb_model.cpython-39.pyc
index 870fbd6..340cd49 100644
Binary files a/models/__pycache__/roma_unsb_model.cpython-39.pyc and b/models/__pycache__/roma_unsb_model.cpython-39.pyc differ
diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py
index 0c88585..d56f29a 100644
--- a/models/roma_unsb_model.py
+++ b/models/roma_unsb_model.py
@@ -405,6 +405,7 @@ class RomaUnsbModel(BaseModel):
             self.mutil_real_A0_tokens = self.netPreViT(real_A0, self.atten_layers, get_tokens=True)
             self.mutil_real_A1_tokens = self.netPreViT(real_A1, self.atten_layers, get_tokens=True)
 
+        print(self.mutil_real_A0_tokens)
         self.mutil_real_A0_tokens = torch.tensor(self.mutil_real_A0_tokens, device=self.device)
         self.mutil_real_A1_tokens = torch.tensor(self.mutil_real_A1_tokens, device=self.device)