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

File "/home/openxs/jj/roma_unsb/models/roma_unsb_model.py", line 421, in forward

Browse Source 
bs = self.mutil_real_A0_tokens.size(0)
AttributeError: 'list' object has no attribute 'size'
This commit is contained in:
bishe 2025-02-23 16:04:38 +08:00
parent d4ba5d8114
commit 102595c040
4 changed files with 5 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

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

1
checkpoints/ROMA_UNSB_001/train_opt.txt
View File

@ -63,6 +63,7 @@ nce_includes_all_negatives_from_minibatch: False
normG: instance
num_patches: 256
num_threads: 4
num_timesteps: 10
output_nc: 3
phase: train
pool_size: 0

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

Binary file not shown.

3
models/roma_unsb_model.py
View File

@ -405,6 +405,9 @@ 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)
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)
# 执行一次SB模块
# ============ 第一步：初始化时间步与时间索引 ============