From 9966211e8b7c7b18cede4b926b90f794a5b82cbb Mon Sep 17 00:00:00 2001 From: bishe <123456789@163.com> Date: Sun, 23 Feb 2025 16:44:28 +0800 Subject: [PATCH] http://47.108.14.56:3000/c/b78fb9d0-d385-4dce-81dd-5173e1cf037a --- checkpoints/ROMA_UNSB_001/loss_log.txt | 1 + .../roma_unsb_model.cpython-39.pyc | Bin 19497 -> 19514 bytes models/networks.py | 2 +- models/roma_unsb_model.py | 1 - 4 files changed, 2 insertions(+), 2 deletions(-) diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt index 95fd4b1..09b323f 100644 --- a/checkpoints/ROMA_UNSB_001/loss_log.txt +++ b/checkpoints/ROMA_UNSB_001/loss_log.txt @@ -4,3 +4,4 @@ ================ 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) ================ +================ Training Loss (Sun Feb 23 16:09:38 2025) ================ diff --git a/models/__pycache__/roma_unsb_model.cpython-39.pyc b/models/__pycache__/roma_unsb_model.cpython-39.pyc index eb910dd05c7b6acaa3053e8e4e775cb8fab26c31..f14a8b5217ca5eb745fe2a036cb09fc119356b33 100644 GIT binary patch delta 3431 zcmcgvZETxY6~5Qsv7N+;?R=+6lct%|d^lbDm8RcWLz2=zjCBi4mgCn;?8J6@{f4d8 zYg|;y2B@pyO3;B&?P{8~At50Th6+RQr+jFLmN8cO2P+c;b=t;<#t=e9oafq0vNZjJ zfJE=(bI&>V+~=P6+~e!#F7hiExief{?PT=#>}%h4?)39O@sR%wOLK}eC2KM)=F%l= zB$Z6-Vu@7ZWO1zJ0nV4DYz_F}q|_yQWIh#(MAPD=yw=)aD;tV$x4viV{sNr727UzH=>QGQN-t$T!@k-zNrbEEiXx8VHi^1+@2KUcikbJ<+~JWVKB zbrFsw#fzByLh*Fqwu$CF*8d>CQM}&&7Uz%39}H&s`Qq4+i?_WCuW5h>aEl;d7bB40 zl9AvLe@9;L_sVC25AZAUc5r!CGi#RfrhJXT;|^;3n(H)(U8s{{ zZZ5E#=|UT46sIfSU0|G;Wogiv-$D!Z6qsgiR3vl!;j0;!;nJ)NmK+!F)AFun2E;a+ zSrf~+4Tt7fr>i#HMs+L8Z`B->BKj{_GBt)Y}i|JC%M)% zF~j3yMvYOI@5L?}nNe$4Y3e=LV?JPbC^L8~Rd_G3XFP@ryH~nC&DzXVRzBoa&INgG zTlLah2Fucm>-Vua6Qyy@giJl{)=B$gp#g*Cy9^Ia8!+k%=1iT+3rJrfopXh>SCRJJ zg|xRq+POm7t4L?T+8JNZy5jaB))gw~wG(;YQ z^w7(s_a6$LV4B;)EbOeAvt<1+%bC-;Ae9IHAEJj`*B0J3t=g#au?>l+my+?TL|1L- zQKs5}n6!3wxXdNAojzIo-ke1oR}#-zQk(Sp_(*<)3bSefJJ>;%9q^z!*h%uRRx`&7 zCbA4yG}~7+6pwPj;)pw7_zPz8+EvkE%>knUy1vkEQc{TcE_KM^g3)d@B3qm^p*_&Z zXLq!putTE9p>fT09zn-w#Nzw3S}J6_Xi=k$T7l=k+zN*O(^fFYGL4mXz%R_Kwu3DC zfudiBPDM>1xY7h{V(@?11keL)BDQHq!Iyvjft9KkFd)l+^D4|&DRk3`RC>)F*qID1Fj})>vXNRe-yywkmVwvU@ z^DZ*Ox@(OUl{TKg&uBJU3Z_ru5FDjvg6j7pWSLO)JBc~;1c@RGP3hB>@#I-khQ3bd zfZ$7|ay(3-Df#^F!!|R%6CLuM-P7H7`@X)+$>jq1oeoIPe!pDWV?^ zi}WGBLU3HTAyxyvB)_$1H2W8b>Jo6b@Na(*UP$=|huTUu zJ(^6Xgz&*&i7abjx^l!4biH{>-Wu(nP?r!(^i$zQu}`_g^k_mnide@0e}K{{!bvR| zT8u43MJ=>3z;iHJ5KKHS!+SUJFUvD~kMbwvPxemN7GSieqk34^g*YQOk9j*jk69-G zKLbnv9s=A4___SpSWo{(8F-(}C4T>mC6SBmc*eAvV<%40{`kEu=ciuYiC3ZRMw$tW*?J7CFzS0k)g!JO8$8Rud+ zF{_0`6NiqAHT#v{IT8r25?rO_oDJ)d`A|A`I$Gwn8H^RL%d%!XFkgf5->_&cNh|Rq zdQ$X4TMv5+qEsYRWvTSmZE}= zQdp0rlI5y5C+H5Y4%+H59Kcj{7Tu7l2?DihD3CHu9WttGY@r(IQkk zL!8x8?)20GoyJfksf+&NHz&Tqd5@f&?63`DflD$u8ROgKjmeMlKDl-3k^P?o<6*$d zfY$)O2dLixs<>w`yasp$@G9V5MC;@7=5&KeB;^}Z+Xj-P546#bpykBUDWNj)CS1P{ lP<7uYO))#~7R%PO4;gOwUC5?ux8xf4hm(gZR^P%0$&qe37yib^S#5M^k5lnFKlh!2f&&V62* zW$8Z%Nc4X1+;h&o=XcM$=e*}D7wAhDs7LYpJp}&l-u$X(EJ9zZO-5dqIu}7xZz?Ek zPOCSiiUmz)$3ewdZM^FUrO$BQ9SftAP}iNQ`C=xO))?bI>TYvbhT5Cme|AJaflkjO z^e@8zO6LLoi9&_jqX~L}k4Dq*e<^W6En@jhZe#2Fz zW_y*G9aU0t87_$239Gx!E>!{_WYLYQLJZG28+w6eU1Lmf942ND%=H++PF@ddZm5!q z?fg1QV4YrbW0g=~R))cPvkwAAt3-9S3z8Lj^jgVlcvWdZs!(MY(f8Qpi0qR3L0MAz0k!}sgsDTywA(Z1$b?}=cRfL zQicaN7$$QzNaKnLrB>Lj2lhwgxIxTbBM8&tMr+kxY7u#{Rc!1P(0P_ghXm>HeMpBI zq&>@|LxOY}S$ndpSvTC?$GYK#tUo|n_ycLA8d|0ruF&HjF@i`5HZo*{8qzlK7=p{* zLE!nz!%q;^ClQI9u~SlL9U&FF_G|)<2mU_=7r3rGcw1G!;SQ5EiHL_%iikwJ*YvQc zw&9u(yPUAN$Rltk+3gjH9S{<&NX1sYCA-7i0SD8)fIAok%P4xlId}l%gc_KmRjkTH zLo>3hf%PC4B#Z0DjY!oFUSka%#*7?f)bX(y25Zzw&`na%$45Qo_U1Dh|ubKw*D&0^&}Z2dZVPvq?Brp>uX@Vx`Vy z^B9=XI<~^v1>=L}l+kH)Rc#-}B6tX{LvWg>!E#8P--mGyT*E+-$TRxk#`xhgwoOC8PIyJzt{mmaM?5@>Gm(+!x`ecSmA&d~?nK{01)gyS<;i z05|qofJd1RML)u)`IX&!%6~#py!h`IeRBm5HKb%bvtm;m)~eo4>dWLSY~d-{5Qiw1o8*ByGgpcNTw!+~8_me`B@ zYkP*nEQ|^%wWKQWagoWxhs;HOcTapld~QhTXOtY>&N;VFcN5Prbb@&5Q)Ir-6ziDN{&k z>@JRM$24Z=m&b=_f&XfJfV#M6uWS#~a4om@G~M_P_{4ndEnN3G1hMbSD7o;-eTN73 zw>^eWVb;CK(kf;>SNG&%Gyk67x_t1TJ zFbJk?MPN6&`vB_hTyaiY%BN2(S|5Ufjd(J9$WDbmA*K~E(lT-4?R7IYie3Y%+^|gbs#65!dXnBq)dTL(Q zGN;lOuRdffc)iFsOvUD#ar{>d8U$&1{ghr{n^7A=dlyPu5W?u>%x#x3qwJ@kig~TC z*EVz%?0Zq#g0K-Gim-`)WvU~x9>rDJy;=MD)GZnsKoh1UGhrj-@tB>x&A)f3Z(HM{ye1ve0FAo zV;BRNygZYkJNWNrK12uk&cjdd`zQtV5Tgk)!2H3TInE(I) diff --git a/models/networks.py b/models/networks.py index 933f792..ebae8aa 100644 --- a/models/networks.py +++ b/models/networks.py @@ -984,7 +984,7 @@ class ResnetGenerator(nn.Module): else: # print("%d: skipping %s %d" % (layer_id, layer.__class__.__name__, feat.size(1))) pass - if layer_id == layers[-1] and encode_only: + if layer_id == len(list(self.model)) - 1 and encode_only: # print('encoder only return features') return feats # return intermediate features alone; stop in the last layers diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py index 0000f0f..11ce30c 100644 --- a/models/roma_unsb_model.py +++ b/models/roma_unsb_model.py @@ -405,7 +405,6 @@ 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.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)