From 102595c0407315f47805296c18815264e16f1685 Mon Sep 17 00:00:00 2001
From: bishe <123456789@163.com>
Date: Sun, 23 Feb 2025 16:04:38 +0800
Subject: [PATCH]   File "/home/openxs/jj/roma_unsb/models/roma_unsb_model.py",
 line 421, in forward     bs = self.mutil_real_A0_tokens.size(0)
 AttributeError: 'list' object has no attribute 'size'

---
 checkpoints/ROMA_UNSB_001/loss_log.txt        |   1 +
 checkpoints/ROMA_UNSB_001/train_opt.txt       |   1 +
 .../roma_unsb_model.cpython-39.pyc            | Bin 19389 -> 19431 bytes
 models/roma_unsb_model.py                     |   3 +++
 4 files changed, 5 insertions(+)

diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt
index a6f9e99..3f966d3 100644
--- a/checkpoints/ROMA_UNSB_001/loss_log.txt
+++ b/checkpoints/ROMA_UNSB_001/loss_log.txt
@@ -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) ================
diff --git a/checkpoints/ROMA_UNSB_001/train_opt.txt b/checkpoints/ROMA_UNSB_001/train_opt.txt
index 18c3b2c..b15424d 100644
--- a/checkpoints/ROMA_UNSB_001/train_opt.txt
+++ b/checkpoints/ROMA_UNSB_001/train_opt.txt
@@ -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                             
diff --git a/models/__pycache__/roma_unsb_model.cpython-39.pyc b/models/__pycache__/roma_unsb_model.cpython-39.pyc
index 1b4d9539fcb1c3b40418c7b8435bb2b4a7ef46d9..870fbd6961245826832fc4e672d8a2f43d994f8f 100644
GIT binary patch
delta 1540
zcmZ{ke{54#6vunJcHP=;nBjgv*aj<Qd$U{1m}Kon9ltiW&f$+GuqLK?>w9ix{gwOP
zq9B3AWNfH|%#*kgiNs-|!e8W(XdoC(l*FHCM8w7)Gw~NjR19b$iSd44Ba1}dAD^6i
z&bjA)@4e@~v$O1nSyn2SlsGNovt2wrT&$lgja0HHjA&&atF$Nq-L9-v{EBVb#w>d*
zI}4Q-WgVTUTIOn<w!-s}MY`KKUv<n@v3p5VQ+g~F(-SFG)77kY8C8i_pJQzltZ8A)
z*je*d0lP;2x=GKIBF3Kryf?81)<J<AFb>h@bpbX@PLIkaDC=ph9EUq1kaMQvYAh-5
zQn|+W7-u}3vFE7Sn-Ymfye;f!<Ggphm3fT9hBj7q95d`XmlJ9JDoUK9`=xU%XjC?e
zqzUTx?O>PaC0{j*8gKjhtZbUf{DW+SlKzt`e))Gzeg~|}0+NHb!v2YZ0YAG!i9jtI
zHx2~)i)wDSq34;%KSbBtHdV}F>azm<+zqn?n4y`r;68p8ws{`6EA*pT2x$a-ARw`v
zJI+;ES7XCm&W=dEd;+l-fz!ZOz}LV6@C|SlI4Y34Gc~3sk}<B5J=DPFsX4Tb9j3h@
zA3IJbLbdFBdM~uv{R&cTxLZC+H$pY+8Cu@n#imGZk8FGaL5HU5vaWMJj!}_3re}m6
zg>Mr>egN1GyaRLtgTPkcUAofl_5GjBKSVgk2BNk^I<E1*F|rP6+)Ck&0p=j;*utJR
zE_KYYRlkdv<mSJk-W1RWoB|v;v^QubT)`IT<?xF29}BN!GjB0xdqf^pbH!P%DlwS6
z2HCeX<ws&dmGM`284IHW_$08nN)e{G1aig6%&<0=QXk9aTg_bhQDg^!3-mx|!2XFa
z*BJXd2d!+57B*G)ncD05HKZH{%nQWc^NXPMayBlf6>i5W1)8ozok=-0q{y-Ejs5@G
zX3u6Ruqbj-nsZ3j<0COGv0Ke+TLsmowt1T0EbS}B_&OF{BCMPB<9eFc!&`><a##((
zQY1N&EwN-q)A&W<l3e+RQI14YeJ!kdU?orocm$&Mtpl+ggE<%G#)gxbAvsA4;d1+@
zqD;V0wp?IKy-3E%Bs#4Y%YanCf1^lO-D)&#{-{i?rfLYzADRTKk!HG9cK(UPVlmFS
zvMLu{F6)U*nr{&9TtQ0w49qvM_(Ga6)qEyySS56|t2}7l$t?@rCBjj+DP7BoqH>b^
z;d1JkQ8lf_;%S}x=$`H)Z2Pe<x*7A*m-p5dHlyfKvi2lc3nhE*vo{LI<MevZ(1!g`
zx)yjJ_zd_KF#kSgOGhz88*+3H$2swJsMtsk_ck(#rh4mZ%^iP-&>6sNn<72K<0H-K
VG=Gym?A_(si--ewjUJ4Q{sYa4loS8}

delta 1532
zcmZ{kZA_b06vulDw9r!2u??8KOiDMNt&UR30t^wP!g!H=5EYu3ClBo-H1sX^d58mM
z_=2LFBf66TA(3q1%l1j<lW5!|nwSucOXdsHq=_$!E}QWM&74sa6Yu{LWLdU;`1PE7
z&OP^kPtWbOIkr5<3N(+$WfPxMtLKN_FwPZp7qPSzF6v=9j%XlSOUuP2?%;7dvmLQL
zVN)Kmt`sk2(?Zz_+fQ0~&^lTEVHUeV1C`U>mqitR6!86zU9gS|`~zc#vb+H{OLbnI
zP0@n4j!n`f?>07JUG;Lt7O3Bs5P=n6ko{!+?b~l>l~$m-o^3fRc4u--uEmo4ZB(*o
zTv=iDR_h*-G)0H~{p>6H%wNU^tTlg+oxMU$H3MvjW^2xs{(=qNz`ekA0maGdVE;&w
zKn?p@>@4)SBbupl-58CV#-wEhdUMO~cA3|i*gHcN4UcZ0Lz$BTy*wYL2Y8h()Hh7>
z-(bsL+^x_xSqNz-fQteO%j8G6u9><z#I^LW(#59`TLRt%z6QPlz6HJm-UAi|GFua)
zW-Kl)LN$%m>@@W^9%A!!rqR#N(&feqc9m{6R_#55)NC`B(D{onJ>tZ;+Nvf~F=H$k
z&XeD2LQC`4D0HBmP1CUh-L0pP>NIpyGfmDX;pJ+hW=aT+@<YPqF9A;j9|0Y}0Pq;_
zF%>oW{CCoSQ-m|DFPt4qMhyNhJo})*84Wk}F$XO-h1fC6{_s3|@^=wa^7*f*Hx2Xv
z7T`cP--p#85O&8?hM|VkXSESswkn*L=6Y1LMvG@Qf^hh!wANf&QikofjqW8;sK}~%
zq=YqNZe+7=Io63SE`dy5JT+vDCiKJU^#PSyp4-sEzo3bhK+fmF_gQbY4A|Kmx!THl
z<YjjA8_1gn<WG%etRI4b&17>e8RgkHCCf0QVOLyB3`RAzqqX<edy~<o1U5x(N^=cr
zW@K13V&nR{ww+KdwVkGmZHn6i{|_8ma3Zwr;C*m+z^w+h2siAE2UWC_UlvP>d;Rrn
zMWPg24r>=s0XzUy(leprl2VwrHFZ_gsIg{3tE{jJVRRwZVz=4)lq|kRqwSSd80Gc0
zNL^B40di%7s$nT~p?#<0PbB8jmG=85<r~<1A$}yuXX1vHFQD8bg=cb;YFr!BxpBuM
z_K3`Iz8oGSt(h?~Ap5Z7GE*aZ(oiEwlkdgCj*X{=rxT!wj-%|UCC5R=s>s<{;doHw
z@I~@>##k-Qc0Qh?h=p;w-Z|L(0#w%mp8#J1-vjc$O3ro|F2;{zYB+kvgQ%!5TI}9V
ui(Q4Rn%?T#T_IcW1LEfaIWxItibsa`C6oLE`m1Zq{X8O0;2k>BJ@OwNdXlgJ

diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py
index 691121d..0c88585 100644
--- a/models/roma_unsb_model.py
+++ b/models/roma_unsb_model.py
@@ -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模块
         
         # ============ 第一步：初始化时间步与时间索引 ============