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

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

in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8688710927963257, max_false: 0.8135088682174683, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9355723857879639, max_false: 0.9302115440368652, correct: True
in: torch.Size([2815, 5120]), out: torch.Size([5120]), max_true: 0.9123719334602356, max_false: 0.9464585185050964, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8145414590835571, max_false: 0.8240596652030945, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.974257230758667, max_false: 0.9508087038993835, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9410850405693054, max_false: 0.9418192505836487, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9441309571266174, max_false: 0.9125450849533081, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9071673154830933, max_false: 0.9082017540931702, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9327974915504456, max_false: 0.9475795030593872, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7190749645233154, max_false: 0.7587426900863647, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8623054027557373, max_false: 0.8588939905166626, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8726115226745605, max_false: 0.8520898222923279, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8443689942359924, max_false: 0.8410443067550659, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9513735771179199, max_false: 0.9599956274032593, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9756813645362854, max_false: 0.8977795839309692, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7046935558319092, max_false: 0.7859461903572083, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9741970300674438, max_false: 0.9586616158485413, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9286705851554871, max_false: 0.8987109065055847, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9762621521949768, max_false: 0.9592257738113403, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9494307041168213, max_false: 0.9432995319366455, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9426623582839966, max_false: 0.9482269287109375, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9448583126068115, max_false: 0.9548918008804321, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9764128923416138, max_false: 0.9468739032745361, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8892486691474915, max_false: 0.8559369444847107, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.86468106508255, max_false: 0.8773118257522583, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6896989345550537, max_false: 0.7235923409461975, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9870611429214478, max_false: 0.9757351279258728, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.856162428855896, max_false: 0.8417779207229614, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9362444877624512, max_false: 0.9287120699882507, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9201620817184448, max_false: 0.9135378003120422, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7417165637016296, max_false: 0.7243208289146423, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9321264624595642, max_false: 0.9292213320732117, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9504503607749939, max_false: 0.9142752289772034, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.7586554288864136, max_false: 0.7543150782585144, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9251052737236023, max_false: 0.8232893347740173, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7807475328445435, max_false: 0.780903697013855, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9136106967926025, max_false: 0.8815542459487915, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8299493789672852, max_false: 0.7767565250396729, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9871114492416382, max_false: 0.9804627895355225, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9746289253234863, max_false: 0.9704639315605164, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.9021334052085876, max_false: 0.7796726226806641, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6944350004196167, max_false: 0.7433002591133118, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9683968424797058, max_false: 0.9460813999176025, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7807927131652832, max_false: 0.8591817617416382, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9697456955909729, max_false: 0.956835150718689, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.978959321975708, max_false: 0.9622810482978821, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9701027870178223, max_false: 0.9311351180076599, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9488490223884583, max_false: 0.8974788784980774, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9649488925933838, max_false: 0.9411239624023438, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9194783568382263, max_false: 0.9042963981628418, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9502086043357849, max_false: 0.8761076331138611, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.8957225680351257, max_false: 0.8459942936897278, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9672424793243408, max_false: 0.911790668964386, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8368499279022217, max_false: 0.842557966709137, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8716603517532349, max_false: 0.7288104891777039, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9426673650741577, max_false: 0.8606019616127014, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9694891571998596, max_false: 0.9507665038108826, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9350309371948242, max_false: 0.9551712870597839, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8105902671813965, max_false: 0.7562464475631714, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9598856568336487, max_false: 0.9429425597190857, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8189600110054016, max_false: 0.7644462585449219, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.821494996547699, max_false: 0.7841824889183044, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.972283124923706, max_false: 0.9574865102767944, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9274049401283264, max_false: 0.9535931944847107, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9680918455123901, max_false: 0.9483453035354614, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9426144957542419, max_false: 0.9606861472129822, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9680326581001282, max_false: 0.9495821595191956, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9321650266647339, max_false: 0.93409264087677, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9666429162025452, max_false: 0.9609996676445007, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7537938356399536, max_false: 0.7111027836799622, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9377580285072327, max_false: 0.9181742668151855, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9697131514549255, max_false: 0.9532234072685242, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9755945801734924, max_false: 0.9549502730369568, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9490609765052795, max_false: 0.9140830636024475, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9703062176704407, max_false: 0.973101019859314, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9334640502929688, max_false: 0.9629010558128357, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.979642391204834, max_false: 0.9843644499778748, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8481565117835999, max_false: 0.8627650141716003, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9630627036094666, max_false: 0.9220961332321167, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8682437539100647, max_false: 0.8686848282814026, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.840857207775116, max_false: 0.7495550513267517, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9009539484977722, max_false: 0.842911422252655, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9581329822540283, max_false: 0.9567884206771851, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8824722766876221, max_false: 0.8898873329162598, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.8550273180007935, max_false: 0.9263612627983093, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9070838093757629, max_false: 0.9174714684486389, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9682331085205078, max_false: 0.9116526246070862, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9459580779075623, max_false: 0.9358164072036743, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9681341052055359, max_false: 0.9658625721931458, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.953453540802002, max_false: 0.9365000128746033, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9180300831794739, max_false: 0.8796574473381042, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.6627174019813538, max_false: 0.8251327872276306, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9863994121551514, max_false: 0.9570512175559998, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9568988084793091, max_false: 0.9261347651481628, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9695934653282166, max_false: 0.957181453704834, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9690336585044861, max_false: 0.9013454914093018, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9749069213867188, max_false: 0.9703136682510376, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9348672032356262, max_false: 0.9271036982536316, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9547266960144043, max_false: 0.9546756744384766, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8208861947059631, max_false: 0.8372678756713867, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9685599207878113, max_false: 0.9663525819778442, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9136372804641724, max_false: 0.907784640789032, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9597740173339844, max_false: 0.8531481027603149, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9007528424263, max_false: 0.8064947724342346, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9557422399520874, max_false: 0.9469412565231323, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9448593258857727, max_false: 0.9464060664176941, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9713718891143799, max_false: 0.9466733932495117, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8197434544563293, max_false: 0.8630071878433228, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9391648173332214, max_false: 0.9272225499153137, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9560624957084656, max_false: 0.9441950917243958, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9844256639480591, max_false: 0.972343921661377, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8373255133628845, max_false: 0.8285642862319946, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9691632986068726, max_false: 0.9694532752037048, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9627476334571838, max_false: 0.9580801725387573, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9462980628013611, max_false: 0.9265073537826538, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9419540762901306, max_false: 0.9135209918022156, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9094734191894531, max_false: 0.8968377709388733, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9714191555976868, max_false: 0.9408824443817139, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9739388823509216, max_false: 0.9842678308486938, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9366143941879272, max_false: 0.9418976902961731, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.939060628414154, max_false: 0.9038357734680176, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9259917736053467, max_false: 0.8904049396514893, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.8557740449905396, max_false: 0.8200932145118713, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9188498258590698, max_false: 0.8465446829795837, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9207420349121094, max_false: 0.9391570091247559, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8895021080970764, max_false: 0.8615888357162476, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9188959002494812, max_false: 0.9040905833244324, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8352975249290466, max_false: 0.8399938941001892, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9602314233779907, max_false: 0.9634405374526978, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8809432983398438, max_false: 0.8569380044937134, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9646783471107483, max_false: 0.9484492540359497, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9307116866111755, max_false: 0.9429051280021667, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9700727462768555, max_false: 0.917935311794281, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.821697473526001, max_false: 0.8137093186378479, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.96458500623703, max_false: 0.8709279894828796, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9600756764411926, max_false: 0.921755313873291, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8686226606369019, max_false: 0.8576404452323914, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8366290330886841, max_false: 0.7975778579711914, correct: True
in: torch.Size([2815, 5120]), out: torch.Size([5120]), max_true: 0.9161838889122009, max_false: 0.935822606086731, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8517867922782898, max_false: 0.8249165415763855, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9505617618560791, max_false: 0.9414021372795105, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8793495297431946, max_false: 0.9339638948440552, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8933183550834656, max_false: 0.9188147783279419, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8800197243690491, max_false: 0.8226775527000427, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.919784665107727, max_false: 0.8834622502326965, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9272797107696533, max_false: 0.9201698899269104, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9711312651634216, max_false: 0.9410573840141296, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9149690866470337, max_false: 0.9309524893760681, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7721612453460693, max_false: 0.7620355486869812, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9621537327766418, max_false: 0.914852499961853, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9700887203216553, max_false: 0.9737173318862915, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9196080565452576, max_false: 0.9333385229110718, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9642385244369507, max_false: 0.9722694754600525, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6839505434036255, max_false: 0.756108820438385, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7972615957260132, max_false: 0.8237475156784058, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.905319094657898, max_false: 0.8701804876327515, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9754443168640137, max_false: 0.9321135878562927, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.956059992313385, max_false: 0.9128386974334717, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9736965894699097, max_false: 0.9376463294029236, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9424849152565002, max_false: 0.9129363894462585, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8036760091781616, max_false: 0.7329485416412354, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9148794412612915, max_false: 0.9056034684181213, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.970917820930481, max_false: 0.9194964170455933, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9573441743850708, max_false: 0.8980706930160522, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9675552845001221, max_false: 0.946611762046814, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8405797481536865, max_false: 0.8680797219276428, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9546180367469788, max_false: 0.8985040187835693, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7370657324790955, max_false: 0.7128033638000488, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9686508178710938, max_false: 0.9634696841239929, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9396368861198425, max_false: 0.9308805465698242, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.922618567943573, max_false: 0.8924145102500916, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.792840838432312, max_false: 0.8102729320526123, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9437822103500366, max_false: 0.9397090673446655, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.7987836003303528, max_false: 0.7031790614128113, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9657257199287415, max_false: 0.9552868604660034, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9242284297943115, max_false: 0.9269343614578247, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9616060256958008, max_false: 0.9518160820007324, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9284944534301758, max_false: 0.9266098141670227, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9209217429161072, max_false: 0.8833626508712769, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9345046281814575, max_false: 0.8769379258155823, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9484780430793762, max_false: 0.9454250931739807, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6937310099601746, max_false: 0.7406700253486633, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9537616968154907, max_false: 0.917684018611908, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9219902753829956, max_false: 0.8932392597198486, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6756916642189026, max_false: 0.5480638146400452, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7800525426864624, max_false: 0.7622833847999573, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9723504185676575, max_false: 0.954246997833252, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.956233024597168, max_false: 0.9330093860626221, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9554522633552551, max_false: 0.9246765375137329, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9218665957450867, max_false: 0.8972891569137573, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.97275710105896, max_false: 0.9740811586380005, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9315486550331116, max_false: 0.9081823229789734, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9496569037437439, max_false: 0.9362796545028687, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8507818579673767, max_false: 0.863134503364563, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8026437163352966, max_false: 0.7981894016265869, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8036575317382812, max_false: 0.9023285508155823, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9444278478622437, max_false: 0.9646324515342712, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9660822749137878, max_false: 0.962459146976471, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9469572305679321, max_false: 0.9054498672485352, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8778159022331238, max_false: 0.8733471632003784, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8498015999794006, max_false: 0.8575224280357361, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9382224678993225, max_false: 0.9362179040908813, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9424105286598206, max_false: 0.9455279111862183, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.7506628036499023, max_false: 0.8158851265907288, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9410052299499512, max_false: 0.9170947670936584, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9516932964324951, max_false: 0.8963344097137451, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9819010496139526, max_false: 0.9704176187515259, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7627677321434021, max_false: 0.7591835856437683, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9445652961730957, max_false: 0.9291329383850098, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8399360179901123, max_false: 0.8128309845924377, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8441792130470276, max_false: 0.8485955595970154, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.701382040977478, max_false: 0.7092835903167725, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.95695561170578, max_false: 0.920218288898468, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9578112363815308, max_false: 0.9594618678092957, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9545429944992065, max_false: 0.9495980143547058, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9720999598503113, max_false: 0.9645562767982483, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9665168523788452, max_false: 0.9148306846618652, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.8910176753997803, max_false: 0.8727468848228455, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9674115180969238, max_false: 0.9144882559776306, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9318803548812866, max_false: 0.8864074945449829, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9267919659614563, max_false: 0.926051676273346, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8910866975784302, max_false: 0.8718671798706055, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8701913356781006, max_false: 0.87728351354599, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.678137481212616, max_false: 0.6528239846229553, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9715285301208496, max_false: 0.9695090055465698, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9591094255447388, max_false: 0.9404572248458862, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.949428915977478, max_false: 0.9451056122779846, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8628436326980591, max_false: 0.8865504264831543, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9705386757850647, max_false: 0.9324935674667358, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.961570143699646, max_false: 0.9326870441436768, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9467605948448181, max_false: 0.8760478496551514, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9460407495498657, max_false: 0.9187461137771606, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9700510501861572, max_false: 0.9408926367759705, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9593390226364136, max_false: 0.9486820697784424, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9536495804786682, max_false: 0.9307403564453125, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7065224051475525, max_false: 0.821350634098053, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9691939949989319, max_false: 0.9496592879295349, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8903759121894836, max_false: 0.8485976457595825, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9243480563163757, max_false: 0.8539035320281982, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8633660674095154, max_false: 0.8828417062759399, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9701662063598633, max_false: 0.9461292028427124, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9361121654510498, max_false: 0.8987518548965454, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9841346740722656, max_false: 0.9549163579940796, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8772192597389221, max_false: 0.8462169170379639, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.9283555746078491, max_false: 0.8965469598770142, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9387666583061218, max_false: 0.9207841157913208, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9809508919715881, max_false: 0.9588504433631897, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8781599998474121, max_false: 0.8393294215202332, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9762299656867981, max_false: 0.95943683385849, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9459957480430603, max_false: 0.9550111889839172, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.958785355091095, max_false: 0.9568904042243958, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9561543464660645, max_false: 0.9470975399017334, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9519657492637634, max_false: 0.9576712846755981, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.915552020072937, max_false: 0.9245922565460205, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9498318433761597, max_false: 0.9417757391929626, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.927651584148407, max_false: 0.9354112148284912, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.8671592473983765, max_false: 0.9659797549247742, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8207208514213562, max_false: 0.8426288366317749, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.971937894821167, max_false: 0.9189800024032593, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9031919240951538, max_false: 0.878522515296936, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9554922580718994, max_false: 0.9557551741600037, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9530515670776367, max_false: 0.9466195702552795, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9690614342689514, max_false: 0.9405999779701233, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8484382629394531, max_false: 0.8940651416778564, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8516057729721069, max_false: 0.8201411366462708, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8972203135490417, max_false: 0.8596870303153992, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9692249894142151, max_false: 0.9068805575370789, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.865614116191864, max_false: 0.8758610486984253, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.5984129905700684, max_false: 0.6162605285644531, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7682442665100098, max_false: 0.7390457987785339, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8693150877952576, max_false: 0.7759641408920288, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9320988655090332, max_false: 0.8939594626426697, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9321409463882446, max_false: 0.9167578816413879, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.898319661617279, max_false: 0.9226083159446716, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9586420655250549, max_false: 0.9214785695075989, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9563065767288208, max_false: 0.9427635073661804, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9485378265380859, max_false: 0.9166743159294128, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9675624370574951, max_false: 0.9131401181221008, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9627736806869507, max_false: 0.9653662443161011, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9003928303718567, max_false: 0.8461095094680786, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9805301427841187, max_false: 0.9771097302436829, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.80584716796875, max_false: 0.8098679780960083, correct: False
in: torch.Size([1023, 5120]), out: torch.Size([5120]), max_true: 0.8110255599021912, max_false: 0.8483085632324219, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9262372851371765, max_false: 0.9372735023498535, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9863170981407166, max_false: 0.9360045194625854, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9233132004737854, max_false: 0.8363483548164368, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9646174907684326, max_false: 0.9104910492897034, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9450038075447083, max_false: 0.8770551085472107, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8758895397186279, max_false: 0.8104268908500671, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.891374945640564, max_false: 0.8958812355995178, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8962990641593933, max_false: 0.8578812479972839, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.741542398929596, max_false: 0.7769705653190613, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9658371210098267, max_false: 0.9601535797119141, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9263997077941895, max_false: 0.9381999969482422, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9579404592514038, max_false: 0.9471668004989624, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9251578450202942, max_false: 0.9292595386505127, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9780584573745728, max_false: 0.9805854558944702, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9332460165023804, max_false: 0.9425255656242371, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9504321813583374, max_false: 0.9166748523712158, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.728711724281311, max_false: 0.7432074546813965, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9159960150718689, max_false: 0.8865713477134705, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9655802249908447, max_false: 0.9410205483436584, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.779685914516449, max_false: 0.6317058205604553, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9161720871925354, max_false: 0.8057777285575867, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9494781494140625, max_false: 0.9233060479164124, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9564393758773804, max_false: 0.9307520389556885, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9721822142601013, max_false: 0.9606581926345825, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8371385335922241, max_false: 0.8711942434310913, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9371464848518372, max_false: 0.9086911082267761, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9380620121955872, max_false: 0.9244282841682434, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.97043776512146, max_false: 0.9370563626289368, correct: True
in: torch.Size([2815, 5120]), out: torch.Size([5120]), max_true: 0.9335798621177673, max_false: 0.9094209671020508, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9558309316635132, max_false: 0.9741588830947876, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6725645661354065, max_false: 0.6915671229362488, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8406689763069153, max_false: 0.7839872241020203, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9571783542633057, max_false: 0.9284479022026062, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9484279751777649, max_false: 0.9370836019515991, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9324597716331482, max_false: 0.9114468097686768, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9816938638687134, max_false: 0.9568327069282532, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9408825635910034, max_false: 0.9004793167114258, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9583169221878052, max_false: 0.9244914054870605, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7568076252937317, max_false: 0.7774184942245483, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.956104040145874, max_false: 0.9215973615646362, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9540529847145081, max_false: 0.9281536936759949, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9795601963996887, max_false: 0.9631317853927612, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9576722383499146, max_false: 0.9219138026237488, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9448756575584412, max_false: 0.9454819560050964, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9255040287971497, max_false: 0.9477382898330688, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9294675588607788, max_false: 0.9143065214157104, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8704460859298706, max_false: 0.9110351800918579, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9483968019485474, max_false: 0.9437980651855469, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7143821120262146, max_false: 0.6948565244674683, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9712374210357666, max_false: 0.9639030694961548, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9363663196563721, max_false: 0.9501520395278931, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9253645539283752, max_false: 0.9368771314620972, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9161278605461121, max_false: 0.9291354417800903, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.9009953737258911, max_false: 0.8910526633262634, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9300130009651184, max_false: 0.9324953556060791, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9577348828315735, max_false: 0.9293605089187622, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8929147720336914, max_false: 0.9025812745094299, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9800743460655212, max_false: 0.9438601732254028, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.7899861931800842, max_false: 0.7641351222991943, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9669348001480103, max_false: 0.9496147632598877, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8501441478729248, max_false: 0.8670936822891235, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8780349493026733, max_false: 0.8787961602210999, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9602978229522705, max_false: 0.9537168741226196, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9679305553436279, max_false: 0.9258891344070435, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8798319101333618, max_false: 0.8325626254081726, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.973929762840271, max_false: 0.9543380737304688, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9462604522705078, max_false: 0.9463866353034973, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9582160115242004, max_false: 0.957882285118103, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.870841920375824, max_false: 0.8609690070152283, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9627039432525635, max_false: 0.9487633109092712, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9206464886665344, max_false: 0.9390557408332825, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9010708928108215, max_false: 0.7553845643997192, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9067805409431458, max_false: 0.8519672751426697, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9510756134986877, max_false: 0.9356060028076172, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9497523903846741, max_false: 0.9044860005378723, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9474785923957825, max_false: 0.9243653416633606, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9625847339630127, max_false: 0.9247856140136719, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9590560793876648, max_false: 0.9283162951469421, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9686679244041443, max_false: 0.9422697424888611, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9572200775146484, max_false: 0.9074649214744568, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9533358812332153, max_false: 0.93110591173172, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9524223208427429, max_false: 0.9327744841575623, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9569039940834045, max_false: 0.9180952310562134, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9183931946754456, max_false: 0.919582724571228, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9497999548912048, max_false: 0.9355225563049316, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9691959023475647, max_false: 0.9450988173484802, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9795969724655151, max_false: 0.949176549911499, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8803082704544067, max_false: 0.8776895999908447, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7230210900306702, max_false: 0.875684916973114, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8519315123558044, max_false: 0.6941365003585815, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9218171238899231, max_false: 0.9227935671806335, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9693033695220947, max_false: 0.913433313369751, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.942897379398346, max_false: 0.9094809293746948, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8560392260551453, max_false: 0.8575301766395569, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8924720287322998, max_false: 0.9104878902435303, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9544935822486877, max_false: 0.9641661047935486, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8393181562423706, max_false: 0.7760310173034668, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9178932905197144, max_false: 0.9041076898574829, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9184995293617249, max_false: 0.8926448822021484, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9397727847099304, max_false: 0.9010230302810669, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9136601090431213, max_false: 0.8360096216201782, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9460651874542236, max_false: 0.9183636903762817, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9678961038589478, max_false: 0.9486934542655945, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9514623880386353, max_false: 0.9604846835136414, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9639101624488831, max_false: 0.9335690140724182, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8624927997589111, max_false: 0.7958398461341858, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.963627815246582, max_false: 0.9642156958580017, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8125868439674377, max_false: 0.8191801905632019, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7310531735420227, max_false: 0.735190749168396, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8883889317512512, max_false: 0.8847535848617554, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9321117401123047, max_false: 0.9229787588119507, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6677985787391663, max_false: 0.685727596282959, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9854074120521545, max_false: 0.950731098651886, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9436105489730835, max_false: 0.927311897277832, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9618930220603943, max_false: 0.9102696180343628, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8448779582977295, max_false: 0.8615816831588745, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9781239628791809, max_false: 0.9722719192504883, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9660548567771912, max_false: 0.9661753177642822, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.9001254439353943, max_false: 0.805399477481842, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8718782663345337, max_false: 0.8385447263717651, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9681535959243774, max_false: 0.9454303979873657, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9042838215827942, max_false: 0.9266799688339233, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9599783420562744, max_false: 0.9000401496887207, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9277765154838562, max_false: 0.9264118075370789, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9735579490661621, max_false: 0.9675260782241821, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9638645648956299, max_false: 0.9487460851669312, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.975836992263794, max_false: 0.9825944900512695, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9500993490219116, max_false: 0.9627060890197754, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8078003525733948, max_false: 0.8535512685775757, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8465135097503662, max_false: 0.8262821435928345, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9097038507461548, max_false: 0.8172733783721924, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.887962818145752, max_false: 0.8703706860542297, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7689723968505859, max_false: 0.7289159893989563, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9298380017280579, max_false: 0.9570268392562866, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9716660380363464, max_false: 0.9447695016860962, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9447042346000671, max_false: 0.9619750380516052, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9879148006439209, max_false: 0.9855644702911377, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9426369667053223, max_false: 0.9455736875534058, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9553865790367126, max_false: 0.9168100953102112, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8978082537651062, max_false: 0.9175082445144653, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9703071117401123, max_false: 0.9791567325592041, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.933040976524353, max_false: 0.9631830453872681, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9498701691627502, max_false: 0.9536970853805542, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7288823127746582, max_false: 0.7318907380104065, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.9496168494224548, max_false: 0.7881041169166565, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8351374268531799, max_false: 0.7075947523117065, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.9058654308319092, max_false: 0.8786374926567078, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9616718888282776, max_false: 0.9682009220123291, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9662949442863464, max_false: 0.932329535484314, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9535174369812012, max_false: 0.903252363204956, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9702566862106323, max_false: 0.9591637253761292, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9472799301147461, max_false: 0.9345799684524536, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8242610692977905, max_false: 0.8116657733917236, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7358879446983337, max_false: 0.7822608351707458, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9724972248077393, max_false: 0.9323288202285767, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.742324709892273, max_false: 0.7272472977638245, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9816293716430664, max_false: 0.947184681892395, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9090297222137451, max_false: 0.9187122583389282, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9031451344490051, max_false: 0.885453462600708, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7079305648803711, max_false: 0.7246999144554138, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9574960470199585, max_false: 0.9146410822868347, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7667188048362732, max_false: 0.8090171217918396, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7878031134605408, max_false: 0.7609114050865173, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.8646900057792664, max_false: 0.9324419498443604, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9740366339683533, max_false: 0.9295629262924194, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9243118166923523, max_false: 0.9228673577308655, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9798999428749084, max_false: 0.9628172516822815, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.968396008014679, max_false: 0.9641481041908264, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9715569615364075, max_false: 0.9439735412597656, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8773189187049866, max_false: 0.8883811235427856, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7328795194625854, max_false: 0.7157273292541504, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9495436549186707, max_false: 0.9406628012657166, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9533238410949707, max_false: 0.9325177669525146, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8712539672851562, max_false: 0.843423068523407, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.9121142029762268, max_false: 0.8808165192604065, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9411145448684692, max_false: 0.8987056612968445, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9626690745353699, max_false: 0.9647526144981384, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9183136820793152, max_false: 0.9182687401771545, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9384048581123352, max_false: 0.9535097479820251, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9074313640594482, max_false: 0.8764582276344299, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9531030654907227, max_false: 0.9623780846595764, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8842067122459412, max_false: 0.898122251033783, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9469320774078369, max_false: 0.9364021420478821, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8836208581924438, max_false: 0.8496628403663635, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9741346836090088, max_false: 0.934712290763855, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9630236625671387, max_false: 0.9322874546051025, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9669713377952576, max_false: 0.8921089172363281, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.8954766988754272, max_false: 0.9356082677841187, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7792165875434875, max_false: 0.8208035826683044, correct: False
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8254075646400452, max_false: 0.8224136829376221, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9520598649978638, max_false: 0.9401158690452576, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9191201329231262, max_false: 0.9305857419967651, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9762455224990845, max_false: 0.9634665250778198, correct: True
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.845142126083374, max_false: 0.8806995749473572, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9774417281150818, max_false: 0.9411947727203369, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9104436039924622, max_false: 0.9347021579742432, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9560196399688721, max_false: 0.9545556902885437, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9522005319595337, max_false: 0.9513881206512451, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9554977416992188, max_false: 0.9612825512886047, correct: False
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9229711890220642, max_false: 0.9156097769737244, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.7169234752655029, max_false: 0.6685668230056763, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.6910861730575562, max_false: 0.7346746921539307, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9733046293258667, max_false: 0.9553401470184326, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9598044753074646, max_false: 0.9346567988395691, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8511683940887451, max_false: 0.8293561339378357, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9470826983451843, max_false: 0.9305485486984253, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.954468309879303, max_false: 0.9737899899482727, correct: False
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9000158309936523, max_false: 0.9022173881530762, correct: False
in: torch.Size([767, 5120]), out: torch.Size([5120]), max_true: 0.8508829474449158, max_false: 0.7846933603286743, correct: True
in: torch.Size([3327, 5120]), out: torch.Size([5120]), max_true: 0.9582399129867554, max_false: 0.9269070029258728, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9522954225540161, max_false: 0.9480915665626526, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9239102005958557, max_false: 0.9201789498329163, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9466436505317688, max_false: 0.9406091570854187, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9395764470100403, max_false: 0.9353918433189392, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9650923609733582, max_false: 0.9515432715415955, correct: True
in: torch.Size([1791, 5120]), out: torch.Size([5120]), max_true: 0.9405072331428528, max_false: 0.9387367367744446, correct: True
in: torch.Size([255, 5120]), out: torch.Size([5120]), max_true: 0.8426257371902466, max_false: 0.8099830150604248, correct: True
---------------------------------------
Summary for Split 'test':
 results: {'correct': {'cat_1': 149, 'cat_2': 202}, 'incorrect': {'cat_1': 101, 'cat_2': 48}}
 accuracy: 70.20%

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