From d2041f4241ea20e25d3f69ad8039faabf0cdb565 Mon Sep 17 00:00:00 2001 From: chingis Date: Sun, 13 Feb 2022 15:40:24 +0900 Subject: [PATCH] fixed input tensor's dimension compatibility issue in accuracy calculation --- main_lincls.py | 2 +- main_moco.py | 2 +- moco/__pycache__/__init__.cpython-38.pyc | Bin 0 -> 148 bytes moco/__pycache__/builder.cpython-38.pyc | Bin 0 -> 4558 bytes moco/__pycache__/loader.cpython-38.pyc | Bin 0 -> 1348 bytes 5 files changed, 2 insertions(+), 2 deletions(-) create mode 100644 moco/__pycache__/__init__.cpython-38.pyc create mode 100644 moco/__pycache__/builder.cpython-38.pyc create mode 100644 moco/__pycache__/loader.cpython-38.pyc diff --git a/main_lincls.py b/main_lincls.py index cb9d51675..942c69744 100755 --- a/main_lincls.py +++ b/main_lincls.py @@ -493,7 +493,7 @@ def accuracy(output, target, topk=(1,)): res = [] for k in topk: - correct_k = correct[:k].view(-1).float().sum(0, keepdim=True) + correct_k = correct[:k].reshape(k*correct.shape[1]).float().sum(0, keepdim=True) res.append(correct_k.mul_(100.0 / batch_size)) return res diff --git a/main_moco.py b/main_moco.py index d7ea97359..defffd537 100755 --- a/main_moco.py +++ b/main_moco.py @@ -393,7 +393,7 @@ def accuracy(output, target, topk=(1,)): res = [] for k in topk: - correct_k = correct[:k].view(-1).float().sum(0, keepdim=True) + correct_k = correct[:k].reshape(k*correct.shape[1]).float().sum(0, keepdim=True) res.append(correct_k.mul_(100.0 / batch_size)) return res diff --git a/moco/__pycache__/__init__.cpython-38.pyc b/moco/__pycache__/__init__.cpython-38.pyc new file mode 100644 index 0000000000000000000000000000000000000000..7cbcaaf04233456486b1f86bd2fa41ec7f1aac68 GIT binary patch literal 148 zcmWIL<>g`kf{%+ilH7sxV-N=!FakLaKwQiMBvKfH88jLFRx%WUgb~CqOZ|-e+*JMK zjLf|B%wm0))Z*-t`~rRF{JfH)%%swi%>2A!{oMTIdZ74M$!T}f+MmStp&8K596n3e5dz=~tQ2onMr^Hoq=p(dm0+10LgXI4G4 z{@5B7sn`@;TuClD?^aT&BDdV~SLB{PCI{k5s*;>il?(a3p4CdWNlqDQ-gM8t*VFUf z@BQAJj}{iH497ozCXKyfJ2=g`*W^8oL zTV~UW+1Q>iVF~*-yTzN1;2*HMa~JmPN<{MDVNWMn7lW8JxnNC0@TMt_e8E~=xWc<# z!J78i!I~9O)oZ%b`J9;7b8k9d5H&roOy`HhqMld9VX<_ZHRrxy7HiI9Kg-H`Cw(m) zWmO-?rD3Nh{LrUcem@m)&%fTux7KlMXBf+yew;+OD^KCZc6@M}kQsD_v0i8^&j;E1 z+FB@Ybgng0*nR%SCoaMY=w~Vndfq7vTEHDcT(kg67yWj=9cLzFwb71Ti zj{JGSx=z>YR=QPTtg>AhJi?Rr|ng~lP$94~51 zm#(*r4fg4Y9~ryeZl$Pn5094$uX}jHIS^7*f+IV>DcE>fSozWMN?{b{#Fzj@`CRu{ z_aUscDs180F|hwtt>@-PyT^qA`LOWrXkAf3{HmkBXkSwFM5n**x8g7#%9ti`l6BI= ze_F(?aM;V&{WE7@s?W6FS})_}XPr^}JwF-tH)H9iEkB9dVcxkGQ-rev)-Rkp_uUuv z?eVOB8WBwJpnTXT)8RnCG-)3M$AJ}2H_o3wzrVtV>wcED@_u;3&*T0emgUa(_j|f= z`o(?qI#+YpWf1kkEQ_$4_ys!9&sxrlPu(LnhRz@pQ&h@y{-paASRyc^2ktC}6L5znac_-{C z=j~1shf-PZ$8Ue6%u#2cs)JC5{WyYR+$l zSsYAXM9x7^9wxCw;s}|U{obJI!b<@L%GLYC8&3z@a9AZMJsbpIv46j`wuSJnMOz4J zCtKTyv+X<`ti6^dx$JBXX>ZvY!JrhI#2>M23~tIrJdHMx`4G&l84kyvHP7b_i!buW z`Qye(UgMs*%$JSP0x?cwFDMa3bMK&>w{gCPGrI(lQ+9I7d|XYs>RfvYxyE;m!svnr zy7n?Fc-InOfUf-zTV}^tVFcFBiZGBh=IHrZ9=x-MUk%=vV4QDh@b9tROtp=t@mqYC!yjC5FrJZvS5Tw)F7caH7%Qk}+eX&N?VpmC zgS#$j9Qz!+gg@gLMlIt4@hrwH9so90`Nc#>R(w;M? zcP!K$<2c*3b7PDeG~r`w!gih9=-Tiks=hGp5VIR&RG$e)y}>VI4?14qj4OpL!2N}X zwJY#wWtA0fi!f@9KAQpW61l+7wm=fS_>|vDrQZ*?iD-wJMyyu20)}YJY@T`M8UNj+ zchd*$4N{r=8(-22l-YfQ{Ig!dQ|PePuR2as0_3R!>l>i5jTRVU(^CkhRNR4g+_o|%A@o;q!o1WRk`7AbOsG=Q1WU@R(>? z2t0ty6g)QC6%*hgxR@F^ZefgVzy>A0Z~+Pr@B%nAX|-pe`6v;oueY&>ivN4yC3U*s zIZECAh|BNO^c51XLewiI&k--nH%L&*Nb;h5odmtV`~itKN!-VR%hLeNf<__uF%-+( zorOR^8zTo)cgLgqH3i z%wWs;@xp{re@VZC39Rm+!YSM~dR4t6ukfI?_z1#|-wvZ=ToTSI*6kh$52}hN$*C)Wes6L4gBDqM7gX!**OY{-i|*G`H{3 zq%QwHy?2NC15fCs?M|MZ@^j>QobRtGJp@2fNJd@~ol2tTE+!gSqY9?cHm2K}X5<2N z>MLbIT_vL@Njyd32?*sTsqUHN8M^gD61wERML#-mC&`BB`~VBQTRdx$BD6=bazvbm zXsb+ArLxR?hwL@7pQh0T5-&mQ@sUL?QzW+)-_gice7kA)fPY%D2Wf+>Cn< zu!{?tz8*@EVcm>+1Xcx(K_s9QRHQ|7#a=Ovn@b><<7U+u&1-U*8r5xS!a@>}NNRti z197>&K&@+n-V&{AH9!*~hCLe32f@xT>`i}hgFvKF5a=#U`%k`2YiQrmdzKC)-^ouP z)WWpsi(=I5;6R!_M=I2@=)Rfukm2)~dEByUk2$>R;yi@&VVp}^=UbaMdY9}K2>me& zu;2FR6RwMriS~_e;N0ap$*h8ZiBBY9-7!%#c^98@<);v}c3?qwbW3CVYAoTZe&+9Y z8Gfq2A?TY2|J1|{WO}V5VuA0A1m!0yNs=n#$DvH9eqy}t>!!PLpma43EV>nDevQX6>3`iPRxk8WQKPg!xcFOIp77vsVbg3%xW7d)Y+{_G~WKx$?s7 zW)cPIpiDGM#=4_FG1_s74dQfFqsn|el|51RTGf3Y$oI&`l$*FTYg1qD6=x^ad{FK= zghmM$>Jfg8Goyx&IaNOYjbkhsHLhz(-7VeVm99?`Nja~acQl?gj!MMpR%5?v=#h4e XccH}oMe=?|FXi!hbLlhosKfsQW%i1) literal 0 HcmV?d00001 diff --git a/moco/__pycache__/loader.cpython-38.pyc b/moco/__pycache__/loader.cpython-38.pyc new file mode 100644 index 0000000000000000000000000000000000000000..a1f4b09f55250757c14608f6f158594ec58ea9a8 GIT binary patch literal 1348 zcmaJ=Pfrvv6mMr{XBSvei6me=^ypr82#KP`C~;LnLWmK(?4{{WyTicD6xtd2ll25w zKY*Sfp7;oUmAQK2=*5F4-ucw=@AuyO^~0@OlLX}Fw?^-mL&y(YTon%% z%P{O?P?AWRkd$^QMcYZ7uCqdpUy-y{0?v{aEvHsf z5vjLvk}I@WW6H3`M$BlCC3Z*OiWRPe|93o~*$P0r1lD?&iMk7` z=zws@c!QkN^>bQI_k>kEABAl>rDs*rH$M2D6Vrg{_foK>Qx$|;aTdikc&+SCu6Kiu z&TY9C+Dx_4xE7-L1PShn;NK=XX-GO^q{1BCEck~7%D`8m_m^xoHt=*F-<}z;Tw-@ z=(>LH6&|Z#vP_Jg&X5?LKe)yxTM+~ouNlJ^SWf2xs8{MV+=-`8C`;mx#JF_`=$GC lb!`X-qf|U9ecjtuVP3h6yRU3KA2Rp_=20PQ(uO