From c0364b0170d24a4a1dd4e44420f22c1d63c136da Mon Sep 17 00:00:00 2001
From: bishe <123456789@163.com>
Date: Sun, 23 Feb 2025 16:09:20 +0800
Subject: [PATCH] debuging

---
 checkpoints/ROMA_UNSB_001/loss_log.txt        |   1 +
 .../roma_unsb_model.cpython-39.pyc            | Bin 19480 -> 19497 bytes
 models/roma_unsb_model.py                     |   4 ++--
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt
index ea5e856..95fd4b1 100644
--- a/checkpoints/ROMA_UNSB_001/loss_log.txt
+++ b/checkpoints/ROMA_UNSB_001/loss_log.txt
@@ -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) ================
diff --git a/models/__pycache__/roma_unsb_model.cpython-39.pyc b/models/__pycache__/roma_unsb_model.cpython-39.pyc
index 340cd490ba975ba42f0f1153108c780419c5079f..eb910dd05c7b6acaa3053e8e4e775cb8fab26c31 100644
GIT binary patch
delta 1766
zcmcgt&u<$=6n<}a<6UoTXKim{J4r)-g|;CjPMSI;wDbp|Nz*f`XhBw`DVDZUP~DZd
z%(4>c4MkmPFFB!I4oGmQge(s9037=lAaUT(8$w7Raic=;-gq&gHL6shdNtlR?|t*;
zo0<2`?(X7eyI8zlEZRW7Kfk-*dGtl`mnPne|7hNLSAv({Mg+g$71^d%68erNp`Fis
z_8uSG;Ee?%>g@zMznoNr(LpgHa(BS5%A7Fku#IcfYLH@J62gtYUOW<(lX0>qgqfnW
zIvT|+zmOEAm10s$A*rVjSkj`I8<ahtObUZA%wmeO(j4}V#whYkXMN<p3&}njYfEC+
zo}rvCq!4L$rx7U6{^URkw7x1+pGgkVM2F(7r3U^Kf7@EX!||`JsqnC{gvn281Y>ek
z7&o;DyT3Dyt(GulsSeU{L6&4WndJ#<AjjkgP5dI$NsdT|7IBpEUMWS62aX(NqRMwn
z7!6Ph_!p{A+b@LP+&?feN;Y^exzN#Q#d`(>71|xUL91%dOGx%hhmiBKlIlS@BdvAs
zk<T8Gt}4>*e~_+bq(ghax;kKOdyliuP=B*lq%+h|(h5(cnQC=F)s67hhtgq6w2<Sn
zn#_jEe3>A}sp+)Y&|bI#qF{gl+d9HlWf3C%_MIl3BtLIaBjny1nG;2scVQ?Ibtu6&
z9mkx^4?%>qWxNBy1Sh4*n|&(nMS6p`6dBzM%EHKaui}@z6Um7bME)idU8cw~Bd88e
zlHC-g4yG)iX69KOcqkw31$dlyUXG_ap;{7Fm6Io(mlMjlLO@F4wMZ)pqIxT;M=_5E
zwNKG|CI0y2zAGkM)GUEQ+u@|IQV+b-VvGve?3Pq-m};H>S+&h)Rc!}?TBcmQdsUsT
zQtk{(SJ8L)Qi<@LW-zLX-H3ACgHO?LuDe8IpM!>_6|ypex~eh029$V))EdXQ)cUe;
z=+095_0LEUW&br=npsnKrz~^jM<ciHKcNKkw(*z}zza`{iaPAiiNaa$9Ig9&av^1z
zoTsv=?&y361&u+pl|@^7qLYMvJz0|tIhks2a}B>i?IIQ9Iz=w3^4lPb%FkP%I}qBJ
z{LOy9?rm&q!IUa+`X7|{r8qhDDSi^Kzw#CS9OLRCyd2N2UNwHCk5TV~)i1S)2mD>q
zEiCjO%|%zUvA2GvsNr<)&9fgN&h;Li`$5O$-ZyLS>v*lVvhffxj2oNZW4ZU+<}CgN
DudT+h

delta 1794
zcmcgt&u<$=6n?Y2vDbENukB52C;b5=rA<?6ry-_IDGiX)P)?kHsK`PZur!g1>aGMr
zv-l7WNGNnAUvk`)DkM08%mw~{s^Ww=AaSTv&Ky7-D<Sc{vFn632niv)8tt3+zJ2q}
z%=?}{-xa&NBL7xCpC$Z0{_)$+4`1bfZHim{Kbmh}Rg;=j1|u)+9ZOR?6Xc99yU)iq
zsX{nm-o-E#cyUQO9U&)V>LUuuIwjLJ+7{~=RZ2vd2BOlxvvj=Y#XjsQnNEauI~HPY
zkc%g^n}~Qip}3k*=xP@$*N{CE*QEmtvzXzmSR=b*F$$j9(OmG{r}z-Q&1$e~A4bk2
z2}Rb;LxPa!X#7}00?g(yIUGNZg`P-ArmKczB;LFo=5$UvTTUdp_xj(p8e*=0zct&N
zldeqjlQ?0~AS@t$V`|r0GOY_W(w<McpiA*Qlc<s|>IslJ#uVcxwTF!zXS~WMg5#m5
zb4=d&PDrOtW>bGb4Yv~HdTL=0CUdY!o%E?=;S~0L5SDRb**bRGUepvH)gF+Gx}4Zy
zX++xU-hwYXCS5V4{l6hy8IkT~$E+)3)>-v1>k;Y!YeRa3I*_!%gEUjEjH&vOxc<KO
zm=ZQ}N>}3fo-v;S<P?UD!>0D>*GNt}<j}S)=vsM+BKyV%P24U&Z(snqzit*}UT1td
zkchdIa0=J4pfd*`dN?vZfpD6WV)4e1O1p_Ssiny2o;8+;oOj*8Q)l8csH@Bt6J17R
znGvXir(ri_p(7F&Rb%97jy=eSvmnKJ7xh$P1GS>5XPms~qMkO+7Jx{hR-hjPF|(GL
zA*S$A`x(4px0zac=u@r~(%Rv)=SE&e?|D}-BXwRDx2!1Lb<C^~HXfC@_a95V@oy#G
zp|CoVxwyHxC-Wn$bHl&L9SGuXj>Vno*8V}<sqP7U`#*?VyFHd{STn`P{lTc}fFk4=
z*Wgeu9vYkxcJQnwSqd&d;HHnE^ORT;(ZuyHACXAv#{GvRLaKLOmd$O)pzxf!h<&fc
z&$Hsx8mgtTV{>x~>*MIkD7vyQIs^0z@r$~yXA<iw7xy~GB~<B4h+Hytc^MY!Qk@1x
zYHtKD59c*?dD{xn2T|nQ4Ik9~M*oYmpNXXZ>bY;kFa6x=oVe0&tZoiIUH#hfZ}Y#F
z?TX3aWBJ0j2CrYpTVi(b&c#oJSQz}icGnimgYVbhv&H6MW8-HbKJ1^o@{{le&MU{n
EpPuo}*8l(j

diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py
index d56f29a..0000f0f 100644
--- a/models/roma_unsb_model.py
+++ b/models/roma_unsb_model.py
@@ -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模块