experiment run details:
  dataset: openworld
  path: /gpfs/mariana/home/envomp/bongard/
  similarity location: final
  prompt structure: batched

---------------------------------------
  test split name: test
---------------------------------------

in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8795895576477051, max_false: 0.8429437875747681, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.808097779750824, max_false: 0.7636058330535889, correct: True
in: torch.Size([88, 5120]), out: torch.Size([5120]), max_true: 0.8999730348587036, max_false: 0.8594987392425537, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8232643008232117, max_false: 0.75785231590271, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9017814993858337, max_false: 0.8615949153900146, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9022261500358582, max_false: 0.846976101398468, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9257253408432007, max_false: 0.9013355374336243, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8666940927505493, max_false: 0.8859790563583374, correct: False
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.8890031576156616, max_false: 0.8492759466171265, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8595855832099915, max_false: 0.8262649774551392, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8617414832115173, max_false: 0.8162114024162292, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8279793858528137, max_false: 0.862588107585907, correct: False
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.943971574306488, max_false: 0.8853567242622375, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8747314214706421, max_false: 0.8183139562606812, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9229341745376587, max_false: 0.8048558235168457, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8956578969955444, max_false: 0.8106762170791626, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8769878149032593, max_false: 0.8545359969139099, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8754356503486633, max_false: 0.8437500596046448, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9461433291435242, max_false: 0.8976592421531677, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8796103596687317, max_false: 0.8645609617233276, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.883639931678772, max_false: 0.8764790892601013, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.822285532951355, max_false: 0.8280699849128723, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9307827353477478, max_false: 0.9206442832946777, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8079261183738708, max_false: 0.8076241612434387, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9185596704483032, max_false: 0.901580274105072, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.7729185819625854, max_false: 0.7616250514984131, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.9504209756851196, max_false: 0.9141635894775391, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9101102352142334, max_false: 0.8739504814147949, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9127403497695923, max_false: 0.8861072063446045, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.8609551787376404, max_false: 0.8674561977386475, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8615231513977051, max_false: 0.8800585865974426, correct: False
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.8319928050041199, max_false: 0.8591902852058411, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9190192222595215, max_false: 0.8352165222167969, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8456515073776245, max_false: 0.7928192019462585, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.67987459897995, max_false: 0.5299339890480042, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8532130718231201, max_false: 0.8282378911972046, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8512581586837769, max_false: 0.8110633492469788, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8571903109550476, max_false: 0.7836233377456665, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9440535306930542, max_false: 0.9356826543807983, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8938178420066833, max_false: 0.8891043066978455, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9247655272483826, max_false: 0.89765864610672, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.816047728061676, max_false: 0.7914615273475647, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9284278154373169, max_false: 0.9198753833770752, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8162766695022583, max_false: 0.8042064905166626, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9342991709709167, max_false: 0.8719257116317749, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.8430692553520203, max_false: 0.7880786657333374, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9183091521263123, max_false: 0.8720666766166687, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8366672396659851, max_false: 0.8217791318893433, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9451102018356323, max_false: 0.886997640132904, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8457326889038086, max_false: 0.8502085208892822, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8971646428108215, max_false: 0.8600285649299622, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.7512340545654297, max_false: 0.721576452255249, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9294794797897339, max_false: 0.9118123054504395, correct: True
in: torch.Size([103, 5120]), out: torch.Size([5120]), max_true: 0.8561172485351562, max_false: 0.8473400473594666, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9414634704589844, max_false: 0.8103040456771851, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8800339698791504, max_false: 0.7628096342086792, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.875229001045227, max_false: 0.8536028861999512, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8542202115058899, max_false: 0.8519361615180969, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9165074825286865, max_false: 0.8151401281356812, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8669614791870117, max_false: 0.8247653841972351, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9232797622680664, max_false: 0.8267901539802551, correct: True
in: torch.Size([110, 5120]), out: torch.Size([5120]), max_true: 0.84742271900177, max_false: 0.7858434319496155, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9180654287338257, max_false: 0.9172163605690002, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.817375659942627, max_false: 0.8148513436317444, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9350919127464294, max_false: 0.9143205285072327, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.8753402233123779, max_false: 0.8794691562652588, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8870521187782288, max_false: 0.852173924446106, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.8624415397644043, max_false: 0.8595506548881531, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9052085876464844, max_false: 0.8728840351104736, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.7420024275779724, max_false: 0.6907362341880798, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8518199920654297, max_false: 0.8000563383102417, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9340826272964478, max_false: 0.890174388885498, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9244958758354187, max_false: 0.9028828144073486, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8934230208396912, max_false: 0.8456544280052185, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.9176881313323975, max_false: 0.912115752696991, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.916920006275177, max_false: 0.9096205830574036, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9264881014823914, max_false: 0.899422287940979, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8195600509643555, max_false: 0.7990754246711731, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9152879118919373, max_false: 0.8102524876594543, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8171546459197998, max_false: 0.7073766589164734, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9402915239334106, max_false: 0.8779921531677246, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.7817440032958984, max_false: 0.7721997499465942, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9026371836662292, max_false: 0.8639860153198242, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8730915188789368, max_false: 0.8713113069534302, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8963237404823303, max_false: 0.780454695224762, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.7253581881523132, max_false: 0.6796807646751404, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9153895378112793, max_false: 0.8014172315597534, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9289790987968445, max_false: 0.8037801384925842, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9009363651275635, max_false: 0.8814551830291748, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8743160963058472, max_false: 0.8218744993209839, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.6615144610404968, max_false: 0.6816566586494446, correct: False
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.8020402193069458, max_false: 0.8280977606773376, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9302331209182739, max_false: 0.8326390385627747, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8887738585472107, max_false: 0.8034152984619141, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.907815158367157, max_false: 0.9242119193077087, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9175440073013306, max_false: 0.8884763717651367, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8994671106338501, max_false: 0.8616340160369873, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.875835657119751, max_false: 0.8617357015609741, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9264480471611023, max_false: 0.9197696447372437, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8943386673927307, max_false: 0.8851171731948853, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9243561029434204, max_false: 0.8566854000091553, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8373038172721863, max_false: 0.7784085869789124, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9004822373390198, max_false: 0.5773842334747314, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.7886564135551453, max_false: 0.5676637887954712, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8802919387817383, max_false: 0.8960971832275391, correct: False
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.88792484998703, max_false: 0.8786144852638245, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9525615572929382, max_false: 0.8700224161148071, correct: True
in: torch.Size([103, 5120]), out: torch.Size([5120]), max_true: 0.8494072556495667, max_false: 0.7934238910675049, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8876164555549622, max_false: 0.8818429112434387, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8990103602409363, max_false: 0.8456554412841797, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9379677176475525, max_false: 0.8959299325942993, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.7833778858184814, max_false: 0.7477639317512512, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8950492143630981, max_false: 0.8813216686248779, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9028143882751465, max_false: 0.8565713763237, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.8935561180114746, max_false: 0.8495071530342102, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.8436757326126099, max_false: 0.8039431571960449, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.93262779712677, max_false: 0.8092353940010071, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.869113564491272, max_false: 0.8095454573631287, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.9494435787200928, max_false: 0.9142431616783142, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.8856249451637268, max_false: 0.8762783408164978, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9220401644706726, max_false: 0.869729220867157, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8493366837501526, max_false: 0.8098072409629822, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.9263508319854736, max_false: 0.8395634293556213, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8353548049926758, max_false: 0.7886864542961121, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.8646640777587891, max_false: 0.8173937201499939, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8623273372650146, max_false: 0.8140543699264526, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9118443131446838, max_false: 0.8728750944137573, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8421283960342407, max_false: 0.8251901268959045, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8989477753639221, max_false: 0.9048181772232056, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9159660935401917, max_false: 0.900485634803772, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.8797047734260559, max_false: 0.8394538164138794, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.7767439484596252, max_false: 0.7564882040023804, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9204214215278625, max_false: 0.8352774977684021, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8798844218254089, max_false: 0.7957109808921814, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.926034688949585, max_false: 0.8199915289878845, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.7962566614151001, max_false: 0.7370219230651855, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9023984670639038, max_false: 0.8518773317337036, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8638482093811035, max_false: 0.872169017791748, correct: False
in: torch.Size([103, 5120]), out: torch.Size([5120]), max_true: 0.8924335241317749, max_false: 0.849672257900238, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.852819561958313, max_false: 0.8179912567138672, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.9303290843963623, max_false: 0.9083446264266968, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8225723505020142, max_false: 0.8206456303596497, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8906720280647278, max_false: 0.8936879634857178, correct: False
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8872592449188232, max_false: 0.807258665561676, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.9170548915863037, max_false: 0.8930495977401733, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8248448371887207, max_false: 0.8181746006011963, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.9253085851669312, max_false: 0.8827166557312012, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8211061954498291, max_false: 0.7939525842666626, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.875253438949585, max_false: 0.8897032737731934, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9054482579231262, max_false: 0.8494171500205994, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.93509840965271, max_false: 0.9349925518035889, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8821578025817871, max_false: 0.8661448955535889, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.9031084179878235, max_false: 0.8744368553161621, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.736976146697998, max_false: 0.7358148694038391, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9067216515541077, max_false: 0.8755878210067749, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9055641889572144, max_false: 0.8850700855255127, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9311080574989319, max_false: 0.881488025188446, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9024925827980042, max_false: 0.8760128617286682, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9215200543403625, max_false: 0.8068882822990417, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.7755523324012756, max_false: 0.721181333065033, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.90626460313797, max_false: 0.8123695254325867, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8638263940811157, max_false: 0.8002122044563293, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.9249101281166077, max_false: 0.8582380414009094, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9159454107284546, max_false: 0.8665264248847961, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9196993112564087, max_false: 0.8997682332992554, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.7961104512214661, max_false: 0.7585252523422241, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9273144602775574, max_false: 0.9048535227775574, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8467251062393188, max_false: 0.8703671097755432, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9412752985954285, max_false: 0.9162035584449768, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.893040657043457, max_false: 0.8807483911514282, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8624207973480225, max_false: 0.8125722408294678, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.887888491153717, max_false: 0.8237010836601257, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.85920250415802, max_false: 0.8354392647743225, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8225809335708618, max_false: 0.7984950542449951, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.912778913974762, max_false: 0.9084046483039856, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8952149152755737, max_false: 0.8787724375724792, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9044348001480103, max_false: 0.9041445255279541, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.7838325500488281, max_false: 0.7669394016265869, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8417552709579468, max_false: 0.8245888948440552, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8721264600753784, max_false: 0.8149791955947876, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9243241548538208, max_false: 0.9095684289932251, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8011648654937744, max_false: 0.8266972899436951, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.913619875907898, max_false: 0.8789320588111877, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8779498934745789, max_false: 0.8338009119033813, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8705171346664429, max_false: 0.831611156463623, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8112053871154785, max_false: 0.8077203631401062, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.9442262053489685, max_false: 0.9125899076461792, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8361092209815979, max_false: 0.8185991048812866, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9422200918197632, max_false: 0.8598597049713135, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8786964416503906, max_false: 0.8297372460365295, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9177275896072388, max_false: 0.9103628396987915, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8557117581367493, max_false: 0.8412746787071228, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.897836446762085, max_false: 0.8357444405555725, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.6205719113349915, max_false: 0.6130905151367188, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9430258274078369, max_false: 0.8600749969482422, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8379209637641907, max_false: 0.7881425619125366, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9057246446609497, max_false: 0.909347414970398, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9164831638336182, max_false: 0.9030380249023438, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.861717700958252, max_false: 0.8786783218383789, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8411731123924255, max_false: 0.8041218519210815, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8740591406822205, max_false: 0.873384952545166, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8687266707420349, max_false: 0.8680784106254578, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8535392880439758, max_false: 0.8559421896934509, correct: False
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.7940632104873657, max_false: 0.7650032043457031, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9387050867080688, max_false: 0.8838836550712585, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9078088402748108, max_false: 0.8239998817443848, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9354196786880493, max_false: 0.9085529446601868, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.7947027087211609, max_false: 0.7928106784820557, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9425336122512817, max_false: 0.8819203972816467, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9037519097328186, max_false: 0.8303349018096924, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9438650608062744, max_false: 0.9115420579910278, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.7247030138969421, max_false: 0.7617716193199158, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9326565265655518, max_false: 0.9009968638420105, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9035525918006897, max_false: 0.9086211919784546, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9269888997077942, max_false: 0.8655989170074463, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8901890516281128, max_false: 0.8582916259765625, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9288026690483093, max_false: 0.8420926928520203, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8339470624923706, max_false: 0.7985185384750366, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8863716721534729, max_false: 0.7937868237495422, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.8671198487281799, max_false: 0.7982903122901917, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9147384166717529, max_false: 0.8725228905677795, correct: True
in: torch.Size([110, 5120]), out: torch.Size([5120]), max_true: 0.8936014771461487, max_false: 0.8485993146896362, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8802425861358643, max_false: 0.8304078578948975, correct: True
in: torch.Size([193, 5120]), out: torch.Size([5120]), max_true: 0.8057956099510193, max_false: 0.7623348236083984, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.9235984086990356, max_false: 0.8907445073127747, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.84164959192276, max_false: 0.8105306625366211, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9363912343978882, max_false: 0.930557906627655, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8560212254524231, max_false: 0.8549753427505493, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9515000581741333, max_false: 0.8651536703109741, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9094087481498718, max_false: 0.8348484039306641, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9085121154785156, max_false: 0.793180525302887, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8702607154846191, max_false: 0.7779123783111572, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9344689249992371, max_false: 0.8887701630592346, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8507722616195679, max_false: 0.810356616973877, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9146443605422974, max_false: 0.8880508542060852, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8084748387336731, max_false: 0.8567100167274475, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9282343983650208, max_false: 0.9050794839859009, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8917860984802246, max_false: 0.8670437932014465, correct: True
in: torch.Size([103, 5120]), out: torch.Size([5120]), max_true: 0.9239684343338013, max_false: 0.9052011966705322, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8786624670028687, max_false: 0.8659891486167908, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9485588669776917, max_false: 0.8754287362098694, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9127532243728638, max_false: 0.838617742061615, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9492682218551636, max_false: 0.9109372496604919, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.8539242148399353, max_false: 0.8309237957000732, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8974770307540894, max_false: 0.8801171779632568, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8684126734733582, max_false: 0.8889955282211304, correct: False
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9521406292915344, max_false: 0.8926798701286316, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8511804342269897, max_false: 0.7940243482589722, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9176570177078247, max_false: 0.8907448053359985, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8569748997688293, max_false: 0.8455652594566345, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.9118485450744629, max_false: 0.8616934418678284, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8414686322212219, max_false: 0.8203575611114502, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9313264489173889, max_false: 0.9342086315155029, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8259291648864746, max_false: 0.8146405220031738, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9221045970916748, max_false: 0.8971403241157532, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8360483050346375, max_false: 0.7963799238204956, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.8813853859901428, max_false: 0.8429666757583618, correct: True
in: torch.Size([110, 5120]), out: torch.Size([5120]), max_true: 0.900798499584198, max_false: 0.8275524377822876, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.9250057935714722, max_false: 0.8825639486312866, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8602824807167053, max_false: 0.8333034515380859, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9060826301574707, max_false: 0.8797990679740906, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9033976197242737, max_false: 0.87800532579422, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.8933154940605164, max_false: 0.8621840476989746, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8151675462722778, max_false: 0.8176373243331909, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9220117330551147, max_false: 0.87226402759552, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8781483173370361, max_false: 0.8333247303962708, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9519853591918945, max_false: 0.8962719440460205, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8561199903488159, max_false: 0.7826141715049744, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.822839081287384, max_false: 0.7165181040763855, correct: True
in: torch.Size([193, 5120]), out: torch.Size([5120]), max_true: 0.8387696146965027, max_false: 0.8077923059463501, correct: True
in: torch.Size([193, 5120]), out: torch.Size([5120]), max_true: 0.9342440962791443, max_false: 0.9125463366508484, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9043241739273071, max_false: 0.8716962933540344, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9100379943847656, max_false: 0.8487557172775269, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8340521454811096, max_false: 0.8465675115585327, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9299941062927246, max_false: 0.8682461977005005, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8943362236022949, max_false: 0.8249903917312622, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.8917266726493835, max_false: 0.8300866484642029, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.863067626953125, max_false: 0.8275991678237915, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9209349155426025, max_false: 0.9026467800140381, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.673378586769104, max_false: 0.692887008190155, correct: False
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.930140495300293, max_false: 0.8906210064888, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8635213971138, max_false: 0.8253467082977295, correct: True
in: torch.Size([73, 5120]), out: torch.Size([5120]), max_true: 0.7409279346466064, max_false: 0.7610963582992554, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9071391224861145, max_false: 0.8652979731559753, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9258855581283569, max_false: 0.841884970664978, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8741816878318787, max_false: 0.769235372543335, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8836774826049805, max_false: 0.8108675479888916, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8798325657844543, max_false: 0.8276938199996948, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9247824549674988, max_false: 0.8801888227462769, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8637896776199341, max_false: 0.843317985534668, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8789919018745422, max_false: 0.8449981808662415, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8201532959938049, max_false: 0.836939811706543, correct: False
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.9353095293045044, max_false: 0.9034605026245117, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8739874362945557, max_false: 0.8554760217666626, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9064889550209045, max_false: 0.9002298712730408, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.8360366821289062, max_false: 0.8127115964889526, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9388600587844849, max_false: 0.9382165670394897, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8966960906982422, max_false: 0.850204586982727, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8824929594993591, max_false: 0.8611551523208618, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8566576838493347, max_false: 0.8641241788864136, correct: False
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9504275321960449, max_false: 0.8680981397628784, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8991833329200745, max_false: 0.820899486541748, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9244158267974854, max_false: 0.8444550037384033, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9100489020347595, max_false: 0.8680741786956787, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9453702569007874, max_false: 0.8920355439186096, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8697401285171509, max_false: 0.832377552986145, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9463530778884888, max_false: 0.8911328911781311, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8993212580680847, max_false: 0.9016205668449402, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8692454099655151, max_false: 0.8099094033241272, correct: True
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.8609315156936646, max_false: 0.7950581312179565, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9170353412628174, max_false: 0.881351113319397, correct: True
in: torch.Size([88, 5120]), out: torch.Size([5120]), max_true: 0.8049378991127014, max_false: 0.7649305462837219, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.8785359859466553, max_false: 0.8838897943496704, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8102540373802185, max_false: 0.804929256439209, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9397271275520325, max_false: 0.90382981300354, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9059023261070251, max_false: 0.8592469692230225, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9271594285964966, max_false: 0.8830876350402832, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8395775556564331, max_false: 0.8036614656448364, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9139066934585571, max_false: 0.8649827837944031, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8444914221763611, max_false: 0.7701519131660461, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9153046607971191, max_false: 0.8863787055015564, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.7960203886032104, max_false: 0.7295776009559631, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9133015871047974, max_false: 0.8279162645339966, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9073780179023743, max_false: 0.8578063249588013, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9036014080047607, max_false: 0.8854317665100098, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9200331568717957, max_false: 0.8758531212806702, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.944179892539978, max_false: 0.8922306895256042, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8814036846160889, max_false: 0.8517001867294312, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9395251274108887, max_false: 0.9348838329315186, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8646252751350403, max_false: 0.8669659495353699, correct: False
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.8952568769454956, max_false: 0.8962095379829407, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.795322597026825, max_false: 0.7707033753395081, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9436275362968445, max_false: 0.9054139256477356, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8312059640884399, max_false: 0.8187147378921509, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9323350191116333, max_false: 0.9089259505271912, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8077932596206665, max_false: 0.7809756994247437, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9117180109024048, max_false: 0.8925796151161194, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9175488948822021, max_false: 0.8679834604263306, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.918536901473999, max_false: 0.8806589841842651, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.7958328127861023, max_false: 0.7916859984397888, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.939156174659729, max_false: 0.8814168572425842, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.7440272569656372, max_false: 0.6879010200500488, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9102086424827576, max_false: 0.8949553370475769, correct: True
in: torch.Size([103, 5120]), out: torch.Size([5120]), max_true: 0.8423382043838501, max_false: 0.8182278871536255, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9159157276153564, max_false: 0.9096866846084595, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8745498657226562, max_false: 0.8523533940315247, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9137322902679443, max_false: 0.904955267906189, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8418673276901245, max_false: 0.841848611831665, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9135475158691406, max_false: 0.8011467456817627, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9161233901977539, max_false: 0.8323063254356384, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8597830533981323, max_false: 0.8124504685401917, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8316486477851868, max_false: 0.7682744860649109, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9542282819747925, max_false: 0.909103512763977, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.828162670135498, max_false: 0.8142157196998596, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8722180128097534, max_false: 0.8428847193717957, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.7445381879806519, max_false: 0.7374069690704346, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8964438438415527, max_false: 0.8809077739715576, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9061899781227112, max_false: 0.8714808821678162, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8685081601142883, max_false: 0.8132879734039307, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9429817199707031, max_false: 0.8348027467727661, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8608176112174988, max_false: 0.7779938578605652, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9397825002670288, max_false: 0.8702083230018616, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9074077606201172, max_false: 0.8221420645713806, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.8841512203216553, max_false: 0.9090040326118469, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9315298199653625, max_false: 0.9136626720428467, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.913712739944458, max_false: 0.8846933841705322, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.887263298034668, max_false: 0.8797475099563599, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.886376678943634, max_false: 0.8795139789581299, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9149761199951172, max_false: 0.8680424094200134, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9398788809776306, max_false: 0.8905224204063416, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.8165023326873779, max_false: 0.8023577332496643, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9121948480606079, max_false: 0.8433578014373779, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8761316537857056, max_false: 0.7626590132713318, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9140813946723938, max_false: 0.8796571493148804, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9065803289413452, max_false: 0.8374971747398376, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8975411653518677, max_false: 0.7745822668075562, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8393354415893555, max_false: 0.7391694784164429, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.919653594493866, max_false: 0.8958132863044739, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.8991640210151672, max_false: 0.8812435865402222, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8664258122444153, max_false: 0.75562584400177, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8242994546890259, max_false: 0.7516372203826904, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8704944849014282, max_false: 0.8487088680267334, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8408814072608948, max_false: 0.8070589303970337, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8682153820991516, max_false: 0.7596227526664734, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.847761332988739, max_false: 0.7963336706161499, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9172102212905884, max_false: 0.9007138013839722, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.862906277179718, max_false: 0.8644599318504333, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9283738136291504, max_false: 0.8309147357940674, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.803514838218689, max_false: 0.7502907514572144, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8872503638267517, max_false: 0.866625964641571, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9107224941253662, max_false: 0.8662102222442627, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9145371913909912, max_false: 0.8510793447494507, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8637565970420837, max_false: 0.8176444172859192, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9359006881713867, max_false: 0.8346914052963257, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.7300469279289246, max_false: 0.6942148804664612, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.930985152721405, max_false: 0.8842430114746094, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8559502959251404, max_false: 0.8427730798721313, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9040067195892334, max_false: 0.8786115646362305, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8498514890670776, max_false: 0.8246859312057495, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9161953330039978, max_false: 0.8740652203559875, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9067788124084473, max_false: 0.8742005825042725, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9234321713447571, max_false: 0.9056732654571533, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.7292523980140686, max_false: 0.7470065951347351, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9309771060943604, max_false: 0.8822824954986572, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.6703681349754333, max_false: 0.7045261859893799, correct: False
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.8889546394348145, max_false: 0.7702752351760864, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8439621329307556, max_false: 0.7812923192977905, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8837223649024963, max_false: 0.8973267078399658, correct: False
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.8538919687271118, max_false: 0.8340646028518677, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9076666831970215, max_false: 0.8948673009872437, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8611218333244324, max_false: 0.8614718914031982, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.880690336227417, max_false: 0.8801276087760925, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8102197051048279, max_false: 0.7921485304832458, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9034981727600098, max_false: 0.8817485570907593, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8123948574066162, max_false: 0.7543676495552063, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8673568367958069, max_false: 0.804686427116394, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.7871624231338501, max_false: 0.8486427068710327, correct: False
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9317004084587097, max_false: 0.8296083807945251, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9102394580841064, max_false: 0.8279316425323486, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9521310925483704, max_false: 0.9368240833282471, correct: True
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.7678765654563904, max_false: 0.7687558531761169, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9371167421340942, max_false: 0.8950550556182861, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8281705975532532, max_false: 0.833486020565033, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8447610139846802, max_false: 0.7890796661376953, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.7155938744544983, max_false: 0.6713247895240784, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9031294584274292, max_false: 0.8783048391342163, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8528064489364624, max_false: 0.805673360824585, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9383238554000854, max_false: 0.9080536365509033, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.7799414396286011, max_false: 0.767585039138794, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9285560846328735, max_false: 0.8989766836166382, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.7932348251342773, max_false: 0.7945809960365295, correct: False
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9398815631866455, max_false: 0.8926489353179932, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8915131092071533, max_false: 0.8602214455604553, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8999444842338562, max_false: 0.9059130549430847, correct: False
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8873777389526367, max_false: 0.8811361789703369, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8602868318557739, max_false: 0.7944061756134033, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.6816350221633911, max_false: 0.6743464469909668, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.9208293557167053, max_false: 0.8226345181465149, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8195409178733826, max_false: 0.7263535261154175, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9357008337974548, max_false: 0.9099619388580322, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8644874691963196, max_false: 0.8834704756736755, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8353797793388367, max_false: 0.7903916239738464, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.7737561464309692, max_false: 0.7273743152618408, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9269705414772034, max_false: 0.9044573307037354, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8977111577987671, max_false: 0.8984951376914978, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9018296599388123, max_false: 0.8506237268447876, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8754319548606873, max_false: 0.8526607155799866, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9207146167755127, max_false: 0.8304705619812012, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8830928206443787, max_false: 0.8587690591812134, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9287581443786621, max_false: 0.8837050795555115, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8868575692176819, max_false: 0.875856876373291, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9488977789878845, max_false: 0.909437894821167, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8642555475234985, max_false: 0.8612964153289795, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8874426484107971, max_false: 0.8215060830116272, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8531813025474548, max_false: 0.8044087886810303, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.9108468294143677, max_false: 0.8557394742965698, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.7676678895950317, max_false: 0.7427147626876831, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8413093686103821, max_false: 0.8495281338691711, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8493369221687317, max_false: 0.8389443159103394, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9496408104896545, max_false: 0.9151126742362976, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8475301265716553, max_false: 0.7894117832183838, correct: True
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.9398285150527954, max_false: 0.8967515230178833, correct: True
in: torch.Size([138, 5120]), out: torch.Size([5120]), max_true: 0.7758036851882935, max_false: 0.753380298614502, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8866238594055176, max_false: 0.8921444416046143, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8457105755805969, max_false: 0.8590865135192871, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9356693029403687, max_false: 0.9028302431106567, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.9111183881759644, max_false: 0.9099767208099365, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.9304537773132324, max_false: 0.8963369727134705, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8920167088508606, max_false: 0.8932492136955261, correct: False
in: torch.Size([178, 5120]), out: torch.Size([5120]), max_true: 0.911374032497406, max_false: 0.8323355317115784, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8766510486602783, max_false: 0.8271276950836182, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8724575042724609, max_false: 0.8923805952072144, correct: False
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.7671953439712524, max_false: 0.8074856400489807, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8835180401802063, max_false: 0.8668552041053772, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.7765052914619446, max_false: 0.7920856475830078, correct: False
in: torch.Size([180, 5120]), out: torch.Size([5120]), max_true: 0.9401432871818542, max_false: 0.8932015895843506, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.7339209318161011, max_false: 0.6964302062988281, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.907846212387085, max_false: 0.8323668837547302, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.872287929058075, max_false: 0.8591486215591431, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9084005355834961, max_false: 0.8758739233016968, correct: True
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.8409426808357239, max_false: 0.8443899750709534, correct: False
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.8958338499069214, max_false: 0.904505729675293, correct: False
in: torch.Size([166, 5120]), out: torch.Size([5120]), max_true: 0.8355982303619385, max_false: 0.8212058544158936, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9177826046943665, max_false: 0.8839367628097534, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.8601268529891968, max_false: 0.8370361924171448, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9069719314575195, max_false: 0.8826842904090881, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8502683639526367, max_false: 0.8181944489479065, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9442258477210999, max_false: 0.8494789004325867, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.8832609057426453, max_false: 0.8117755651473999, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.9260860085487366, max_false: 0.8919001817703247, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8915287256240845, max_false: 0.8535202145576477, correct: True
in: torch.Size([118, 5120]), out: torch.Size([5120]), max_true: 0.8515214323997498, max_false: 0.8307732343673706, correct: True
in: torch.Size([163, 5120]), out: torch.Size([5120]), max_true: 0.8682925701141357, max_false: 0.8217747211456299, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9019648432731628, max_false: 0.8845495581626892, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8577223420143127, max_false: 0.8370664715766907, correct: True
in: torch.Size([133, 5120]), out: torch.Size([5120]), max_true: 0.9057819843292236, max_false: 0.8553487062454224, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.7737758755683899, max_false: 0.7972336411476135, correct: False
in: torch.Size([148, 5120]), out: torch.Size([5120]), max_true: 0.9471265077590942, max_false: 0.9120804071426392, correct: True
in: torch.Size([152, 5120]), out: torch.Size([5120]), max_true: 0.8926428556442261, max_false: 0.8836492896080017, correct: True
in: torch.Size([208, 5120]), out: torch.Size([5120]), max_true: 0.9460322856903076, max_false: 0.9145514965057373, correct: True
---------------------------------------
Summary for Split 'test':
 results: {'correct': {'cat_1': 212, 'cat_2': 229}, 'incorrect': {'cat_1': 38, 'cat_2': 21}}
 accuracy: 88.20%

---------------------------------------
