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

debuging

Browse Source
This commit is contained in:
bishe 2025-02-23 16:09:20 +08:00
parent 67b18f082c
commit c0364b0170
3 changed files with 3 additions and 2 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

@ -3,3 +3,4 @@
================ 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) ================
================ Training Loss (Sun Feb 23 16:06:44 2025) ================

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

Binary file not shown.

4
models/roma_unsb_model.py
View File

@ -406,8 +406,8 @@ class RomaUnsbModel(BaseModel):
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)
self.mutil_real_A0_tokens = torch.cat(self.mutil_real_A0_tokens, dim=1).to(self.device)
self.mutil_real_A1_tokens = torch.cat(self.mutil_real_A1_tokens, dim=1).to(self.device)
# 执行一次SB模块