123456/roma_unsb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

debug

Browse Source
This commit is contained in:
bishe 2025-02-23 16:06:28 +08:00
parent 102595c040
commit 67b18f082c
3 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
checkpoints/ROMA_UNSB_001/loss_log.txt
View File

@ -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) ================

BIN
models/__pycache__/roma_unsb_model.cpython-39.pyc
View File

Binary file not shown.

1
models/roma_unsb_model.py
View File

@ -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)