experiment run details:
  dataset: openworld
  model: pixtral hoi lora
  path: /gpfs/mariana/home/envomp/bongard/
---------------------------------------
  test split name: test
---------------------------------------

in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9046370983123779, max_false: 0.8730948567390442, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8802478313446045, max_false: 0.8446768522262573, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.9210811257362366, max_false: 0.897369921207428, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8580679893493652, max_false: 0.8191468715667725, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9209156632423401, max_false: 0.8905826210975647, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9315593242645264, max_false: 0.8691304326057434, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9405049085617065, max_false: 0.9103716015815735, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8906749486923218, max_false: 0.91944819688797, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9139372706413269, max_false: 0.8909800052642822, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8881176710128784, max_false: 0.861912727355957, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8864858150482178, max_false: 0.8427132368087769, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8384172320365906, max_false: 0.8780364394187927, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9477531909942627, max_false: 0.8775951266288757, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9038594961166382, max_false: 0.8534035682678223, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9353539943695068, max_false: 0.8408865928649902, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9192688465118408, max_false: 0.8334052562713623, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9329389333724976, max_false: 0.9075185060501099, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8964202404022217, max_false: 0.8687169551849365, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9111458659172058, max_false: 0.8675228357315063, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9180903434753418, max_false: 0.9106849431991577, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8797395825386047, max_false: 0.8780253529548645, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8298421502113342, max_false: 0.8286386728286743, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.947256326675415, max_false: 0.9347244501113892, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8589519262313843, max_false: 0.8582903742790222, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9263119697570801, max_false: 0.9081799983978271, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8280749917030334, max_false: 0.8156840801239014, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9567859172821045, max_false: 0.9199512004852295, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9385526180267334, max_false: 0.904602587223053, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9450494050979614, max_false: 0.9014257192611694, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8729942440986633, max_false: 0.8922275900840759, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8631471991539001, max_false: 0.896952748298645, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8710322976112366, max_false: 0.8886775970458984, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9310128092765808, max_false: 0.8704289793968201, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.833812415599823, max_false: 0.7946990728378296, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.7523095607757568, max_false: 0.6450706720352173, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.879120945930481, max_false: 0.8704681396484375, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8847029209136963, max_false: 0.824868381023407, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8675599694252014, max_false: 0.8077932000160217, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9607168436050415, max_false: 0.9472017884254456, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8979490995407104, max_false: 0.8925754427909851, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9451034665107727, max_false: 0.9001263976097107, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8459035754203796, max_false: 0.8416312336921692, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9392633438110352, max_false: 0.9176103472709656, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8657748699188232, max_false: 0.8607743382453918, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9536864757537842, max_false: 0.8808805346488953, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8678419589996338, max_false: 0.8255627751350403, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.936392068862915, max_false: 0.8937666416168213, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8761647343635559, max_false: 0.8541007041931152, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9492580890655518, max_false: 0.887321949005127, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8846746683120728, max_false: 0.8869160413742065, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9113973379135132, max_false: 0.8916749954223633, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8045989274978638, max_false: 0.7699882388114929, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9306795001029968, max_false: 0.9229676723480225, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.8667351007461548, max_false: 0.8585954904556274, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9673178791999817, max_false: 0.8662980198860168, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9163367748260498, max_false: 0.8098294734954834, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.906697154045105, max_false: 0.8722646236419678, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8925569653511047, max_false: 0.8861814737319946, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9351643323898315, max_false: 0.868212878704071, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9107950925827026, max_false: 0.8868635892868042, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9437955617904663, max_false: 0.8813605308532715, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.8861756920814514, max_false: 0.8255798816680908, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9390024542808533, max_false: 0.9331055879592896, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.85821533203125, max_false: 0.8618203997612, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9441192150115967, max_false: 0.9412597417831421, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8997427821159363, max_false: 0.9042027592658997, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9005903601646423, max_false: 0.8581762313842773, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8986599445343018, max_false: 0.899555504322052, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9201739430427551, max_false: 0.9005731344223022, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8734146952629089, max_false: 0.8569913506507874, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8866860866546631, max_false: 0.8496859073638916, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.958164632320404, max_false: 0.9042829871177673, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9293504357337952, max_false: 0.9134786128997803, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.916994571685791, max_false: 0.8799751996994019, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9415034055709839, max_false: 0.9379333257675171, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9397271275520325, max_false: 0.9436952471733093, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9407413005828857, max_false: 0.9354491233825684, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8875970244407654, max_false: 0.84898841381073, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9250243902206421, max_false: 0.8395587205886841, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8333525657653809, max_false: 0.7466093301773071, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9524505734443665, max_false: 0.8882251977920532, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8211217522621155, max_false: 0.8151931166648865, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9225674867630005, max_false: 0.892975389957428, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9015947580337524, max_false: 0.9063367247581482, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9280073642730713, max_false: 0.8360357284545898, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8498477339744568, max_false: 0.8138638734817505, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9121009111404419, max_false: 0.8112940192222595, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.941417932510376, max_false: 0.8204090595245361, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9141989946365356, max_false: 0.8806183338165283, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9013435244560242, max_false: 0.856454610824585, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8333641886711121, max_false: 0.7882192730903625, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.180461585521698, max_false: 0.169168159365654, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9358311295509338, max_false: 0.834311306476593, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8998145461082458, max_false: 0.8268834352493286, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9323392510414124, max_false: 0.9472903609275818, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9483643770217896, max_false: 0.909229576587677, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9092305898666382, max_false: 0.8860066533088684, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9063907861709595, max_false: 0.898652970790863, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9419266581535339, max_false: 0.9383656978607178, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9205694198608398, max_false: 0.913800835609436, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9509000182151794, max_false: 0.8979640007019043, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8692971467971802, max_false: 0.8269250392913818, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9541670680046082, max_false: 0.7288374304771423, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8242002129554749, max_false: 0.6489099264144897, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.900561511516571, max_false: 0.9012923836708069, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9077804088592529, max_false: 0.9160998463630676, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.951388955116272, max_false: 0.8849742412567139, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.8823951482772827, max_false: 0.8480556011199951, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9294036030769348, max_false: 0.8880810737609863, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9151668548583984, max_false: 0.8507442474365234, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.94442218542099, max_false: 0.9113119840621948, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8416550755500793, max_false: 0.838961124420166, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9213453531265259, max_false: 0.9031442403793335, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9393019080162048, max_false: 0.9073633551597595, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9174366593360901, max_false: 0.8737305998802185, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8646350502967834, max_false: 0.8243361115455627, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9382879734039307, max_false: 0.8313839435577393, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8942561745643616, max_false: 0.8543899059295654, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.9589275121688843, max_false: 0.9298610091209412, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9213132858276367, max_false: 0.908475399017334, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9443631768226624, max_false: 0.9142810106277466, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8859726786613464, max_false: 0.8461630940437317, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9409734606742859, max_false: 0.8542823195457458, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8566576838493347, max_false: 0.8228110671043396, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.863551914691925, max_false: 0.8295636177062988, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8772624135017395, max_false: 0.844039261341095, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9292877912521362, max_false: 0.916968584060669, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8892455101013184, max_false: 0.8872358202934265, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9441666007041931, max_false: 0.9342295527458191, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9369476437568665, max_false: 0.9293824434280396, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8852339386940002, max_false: 0.8453847169876099, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8167014718055725, max_false: 0.8179923892021179, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9367587566375732, max_false: 0.8622034192085266, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9268338084220886, max_false: 0.8855403661727905, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9413007497787476, max_false: 0.8571779131889343, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8397953510284424, max_false: 0.8066397309303284, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9342875480651855, max_false: 0.8711272478103638, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8781703114509583, max_false: 0.9119371175765991, correct: False
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9242146015167236, max_false: 0.8912498950958252, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8916784524917603, max_false: 0.8488089442253113, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.936771810054779, max_false: 0.9264261722564697, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.857614278793335, max_false: 0.8560914397239685, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9225784540176392, max_false: 0.901382565498352, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.932981550693512, max_false: 0.8683785200119019, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9268350601196289, max_false: 0.9169196486473083, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8707678318023682, max_false: 0.8640556931495667, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9446531534194946, max_false: 0.9182971119880676, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8711755871772766, max_false: 0.846812903881073, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8923267126083374, max_false: 0.9008030891418457, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9267418384552002, max_false: 0.8831011652946472, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9485130906105042, max_false: 0.9543286561965942, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9109775424003601, max_false: 0.8899693489074707, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9288479089736938, max_false: 0.9007789492607117, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7889653444290161, max_false: 0.7881292104721069, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9226071238517761, max_false: 0.9086025357246399, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9203254580497742, max_false: 0.9013651609420776, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9457730650901794, max_false: 0.9178922176361084, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9212878346443176, max_false: 0.8970875144004822, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9311841130256653, max_false: 0.8405966758728027, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8305354714393616, max_false: 0.7521040439605713, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9477567076683044, max_false: 0.8896956443786621, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8828964233398438, max_false: 0.8431529998779297, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.946027934551239, max_false: 0.8999473452568054, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9163828492164612, max_false: 0.8847134113311768, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9257152080535889, max_false: 0.925958514213562, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8703397512435913, max_false: 0.8487272262573242, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9489149451255798, max_false: 0.9218807816505432, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8760193586349487, max_false: 0.8959096074104309, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9535820484161377, max_false: 0.9338435530662537, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9402757883071899, max_false: 0.9361996650695801, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8870797157287598, max_false: 0.8334898352622986, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9030031561851501, max_false: 0.8516852259635925, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8662987947463989, max_false: 0.8644601106643677, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8534628748893738, max_false: 0.8003175258636475, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9287024736404419, max_false: 0.9199485182762146, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9255638718605042, max_false: 0.9225248694419861, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9323205351829529, max_false: 0.9227139353752136, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8837255239486694, max_false: 0.8744605779647827, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8803839087486267, max_false: 0.8715669512748718, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9101895689964294, max_false: 0.8692108988761902, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9309298396110535, max_false: 0.9184572696685791, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8689586520195007, max_false: 0.9019773006439209, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.930941104888916, max_false: 0.9010379910469055, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8930197358131409, max_false: 0.8576163649559021, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9022530913352966, max_false: 0.8114062547683716, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.17447549104690552, max_false: 0.1667417585849762, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9618603587150574, max_false: 0.9171420335769653, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8716409802436829, max_false: 0.8512366414070129, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9463282823562622, max_false: 0.893007218837738, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8829246759414673, max_false: 0.8225839734077454, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9544863700866699, max_false: 0.9493110775947571, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8975361585617065, max_false: 0.8826200366020203, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9206162095069885, max_false: 0.8784367442131042, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6841077208518982, max_false: 0.676463782787323, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9532307386398315, max_false: 0.8738791942596436, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8504336476325989, max_false: 0.8152318000793457, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9260451793670654, max_false: 0.9383761882781982, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9323930144309998, max_false: 0.9301873445510864, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9032936692237854, max_false: 0.9019365906715393, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8567472100257874, max_false: 0.8324816823005676, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8983060121536255, max_false: 0.9205383062362671, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9040837287902832, max_false: 0.8948667049407959, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8926541209220886, max_false: 0.8848413228988647, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8173860311508179, max_false: 0.8421738743782043, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9475780129432678, max_false: 0.8834471106529236, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9398777484893799, max_false: 0.8841426968574524, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9621217250823975, max_false: 0.9318936467170715, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8532347679138184, max_false: 0.8494300246238708, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9610872864723206, max_false: 0.9113103151321411, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9114372134208679, max_false: 0.8493696451187134, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.957089364528656, max_false: 0.9339851140975952, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.776311457157135, max_false: 0.7881355881690979, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9501767754554749, max_false: 0.9220812320709229, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9315919280052185, max_false: 0.9275529384613037, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9366620182991028, max_false: 0.9099432826042175, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9275586605072021, max_false: 0.898098886013031, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9365420341491699, max_false: 0.8852888345718384, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8862994909286499, max_false: 0.8410634994506836, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9193647503852844, max_false: 0.8433716893196106, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9023662209510803, max_false: 0.8438547253608704, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9326582551002502, max_false: 0.8974462747573853, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.9243113994598389, max_false: 0.8875728845596313, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9146297574043274, max_false: 0.8781543374061584, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.8395780324935913, max_false: 0.8109512329101562, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9292721152305603, max_false: 0.9235056042671204, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8701128363609314, max_false: 0.8378664255142212, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9484707713127136, max_false: 0.948041558265686, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9015891551971436, max_false: 0.8994343876838684, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9577571749687195, max_false: 0.8877010941505432, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9467551112174988, max_false: 0.884070634841919, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9320664405822754, max_false: 0.8390213251113892, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8883855938911438, max_false: 0.8036632537841797, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9539490938186646, max_false: 0.9284568428993225, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8967217803001404, max_false: 0.8556742072105408, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9307175874710083, max_false: 0.9118956327438354, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8684290647506714, max_false: 0.9021455645561218, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9506921768188477, max_false: 0.9255353212356567, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9234203100204468, max_false: 0.9078454375267029, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9423282146453857, max_false: 0.9068320393562317, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8806143403053284, max_false: 0.8463767170906067, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9571353197097778, max_false: 0.9090283513069153, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9376132488250732, max_false: 0.8654496073722839, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9499849081039429, max_false: 0.9112416505813599, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8766265511512756, max_false: 0.8620237112045288, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9015083312988281, max_false: 0.8973468542098999, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8839634656906128, max_false: 0.8927693367004395, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.959979772567749, max_false: 0.9240710735321045, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.914201021194458, max_false: 0.8666883707046509, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9416071772575378, max_false: 0.9099951982498169, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8960062265396118, max_false: 0.8814494013786316, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9251840710639954, max_false: 0.8769891858100891, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8798363208770752, max_false: 0.8594826459884644, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.93715900182724, max_false: 0.9463530778884888, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8775551915168762, max_false: 0.8609545826911926, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9439048171043396, max_false: 0.9125556945800781, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8401030898094177, max_false: 0.804236650466919, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8980510830879211, max_false: 0.868851900100708, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.922433078289032, max_false: 0.8457881212234497, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.946147620677948, max_false: 0.9133915901184082, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.898714005947113, max_false: 0.8796283602714539, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9262209534645081, max_false: 0.8940345048904419, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9147608280181885, max_false: 0.8883563876152039, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9072543978691101, max_false: 0.8761875629425049, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8312998414039612, max_false: 0.8574634194374084, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9484864473342896, max_false: 0.9067444801330566, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9164678454399109, max_false: 0.8742610216140747, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9496566653251648, max_false: 0.9045200943946838, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9233903288841248, max_false: 0.870498538017273, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8584611415863037, max_false: 0.775159478187561, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.8823323249816895, max_false: 0.8458457589149475, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.9588496088981628, max_false: 0.9345902800559998, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9412232041358948, max_false: 0.8995779156684875, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9321208000183105, max_false: 0.877267599105835, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8865777254104614, max_false: 0.8945263028144836, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9320242404937744, max_false: 0.9102563261985779, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9127863049507141, max_false: 0.8640270233154297, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9277207851409912, max_false: 0.8660722970962524, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9180097579956055, max_false: 0.8855498433113098, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9487828016281128, max_false: 0.924736738204956, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.7670609951019287, max_false: 0.7789078950881958, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9528183937072754, max_false: 0.9338117837905884, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8954098224639893, max_false: 0.8618575930595398, correct: True
in: torch.Size([74, 5120]), out: torch.Size([5120]), max_true: 0.8914084434509277, max_false: 0.8726891279220581, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9230055809020996, max_false: 0.8895711898803711, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9516140818595886, max_false: 0.8888956904411316, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9043070077896118, max_false: 0.8253225088119507, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.925567626953125, max_false: 0.8616107702255249, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9107290506362915, max_false: 0.8426975011825562, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9394345879554749, max_false: 0.8964729905128479, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9017151594161987, max_false: 0.8873669505119324, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9114904999732971, max_false: 0.8826761841773987, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8510562181472778, max_false: 0.8780129551887512, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9310299754142761, max_false: 0.9002730846405029, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9277453422546387, max_false: 0.9182922840118408, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9155783653259277, max_false: 0.9051425457000732, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8595672249794006, max_false: 0.8607583045959473, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9418227076530457, max_false: 0.9288133382797241, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9274042844772339, max_false: 0.8946741819381714, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9144906401634216, max_false: 0.902247428894043, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8914225697517395, max_false: 0.8968380689620972, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9417822957038879, max_false: 0.8526704907417297, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9210445284843445, max_false: 0.8505207896232605, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9346492290496826, max_false: 0.866290271282196, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9211545586585999, max_false: 0.8989723324775696, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9331587553024292, max_false: 0.8914801478385925, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9115239381790161, max_false: 0.8728313446044922, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9556270837783813, max_false: 0.9187788963317871, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.923949658870697, max_false: 0.9109141826629639, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8903248906135559, max_false: 0.843889594078064, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8972728252410889, max_false: 0.8230359554290771, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9253875613212585, max_false: 0.8886404633522034, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.8694831728935242, max_false: 0.8384718894958496, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9191128015518188, max_false: 0.913844645023346, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8555067777633667, max_false: 0.8556323051452637, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9498850107192993, max_false: 0.9295588135719299, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9334113001823425, max_false: 0.9139239192008972, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9497734904289246, max_false: 0.8992825150489807, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8947808146476746, max_false: 0.8746355175971985, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9544928073883057, max_false: 0.8686664700508118, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9029654264450073, max_false: 0.8378831148147583, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9312683939933777, max_false: 0.8688764572143555, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8357603549957275, max_false: 0.7914043068885803, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9451751708984375, max_false: 0.8891710042953491, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9337620139122009, max_false: 0.8878031373023987, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9263213872909546, max_false: 0.9143154621124268, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9347217082977295, max_false: 0.8743718862533569, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9624727368354797, max_false: 0.9064667820930481, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8965860605239868, max_false: 0.8858134150505066, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.938092827796936, max_false: 0.9317106008529663, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9132528901100159, max_false: 0.9129660725593567, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9120624661445618, max_false: 0.9158073663711548, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8419473767280579, max_false: 0.8137471079826355, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9473235607147217, max_false: 0.9123201370239258, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8718269467353821, max_false: 0.8630380630493164, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9398130774497986, max_false: 0.9264272451400757, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8627956509590149, max_false: 0.8489091992378235, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9462355971336365, max_false: 0.9114640355110168, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9331038594245911, max_false: 0.9096720814704895, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.946518063545227, max_false: 0.9247186779975891, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8630495667457581, max_false: 0.8531462550163269, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9458970427513123, max_false: 0.8840131759643555, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.848695695400238, max_false: 0.8208296895027161, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9415464997291565, max_false: 0.9301755428314209, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.8853124976158142, max_false: 0.864709734916687, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9395527839660645, max_false: 0.928113579750061, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9086266160011292, max_false: 0.8893627524375916, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9334567785263062, max_false: 0.9295720458030701, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.891275942325592, max_false: 0.8827219009399414, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.919998049736023, max_false: 0.8221556544303894, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9279195666313171, max_false: 0.8793935179710388, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8997066020965576, max_false: 0.8664669394493103, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8873164653778076, max_false: 0.8242176175117493, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9479602575302124, max_false: 0.9337145686149597, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8908066749572754, max_false: 0.8622211813926697, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8986493349075317, max_false: 0.8658031225204468, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7833349108695984, max_false: 0.7792790532112122, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9177320599555969, max_false: 0.9159308671951294, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9291405081748962, max_false: 0.8925266861915588, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.917432427406311, max_false: 0.8728776574134827, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9536396265029907, max_false: 0.856964647769928, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8953357338905334, max_false: 0.8400878310203552, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9390859007835388, max_false: 0.9007918834686279, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9383764266967773, max_false: 0.8653036952018738, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9300601482391357, max_false: 0.9110719561576843, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9375743865966797, max_false: 0.9121549129486084, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9161345362663269, max_false: 0.8959531188011169, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9136732220649719, max_false: 0.8999297022819519, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9302083849906921, max_false: 0.8950331211090088, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9419979453086853, max_false: 0.896236002445221, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9431113600730896, max_false: 0.9109190106391907, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8756077289581299, max_false: 0.8991364240646362, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9232335686683655, max_false: 0.8477782607078552, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9052307605743408, max_false: 0.8159016370773315, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.932042121887207, max_false: 0.9070585370063782, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9368144273757935, max_false: 0.8807641267776489, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.914056122303009, max_false: 0.830651044845581, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8859301805496216, max_false: 0.7947589755058289, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9363825917243958, max_false: 0.9037554860115051, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8983967900276184, max_false: 0.8920637965202332, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8845657706260681, max_false: 0.7853266000747681, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8595712184906006, max_false: 0.7927134037017822, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8958905339241028, max_false: 0.8671563863754272, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8736903667449951, max_false: 0.8466922044754028, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.896869957447052, max_false: 0.8020457625389099, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8868411779403687, max_false: 0.8287975192070007, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9544190168380737, max_false: 0.9252119064331055, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9065313935279846, max_false: 0.9355906248092651, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9503233432769775, max_false: 0.860135555267334, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8425649404525757, max_false: 0.8018569350242615, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9076520800590515, max_false: 0.8703717589378357, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9397934079170227, max_false: 0.8888928890228271, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9350588917732239, max_false: 0.9023686051368713, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9019576907157898, max_false: 0.855230450630188, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.908026933670044, max_false: 0.8003854155540466, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7967275381088257, max_false: 0.7915647625923157, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9505274295806885, max_false: 0.9211750626564026, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9107784032821655, max_false: 0.8906974196434021, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9180105924606323, max_false: 0.8911709189414978, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.903327465057373, max_false: 0.9001551270484924, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9431939125061035, max_false: 0.917692244052887, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9251754879951477, max_false: 0.8987435102462769, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9390484094619751, max_false: 0.9127438068389893, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8254522681236267, max_false: 0.8287887573242188, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9287545084953308, max_false: 0.9003214836120605, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8492652177810669, max_false: 0.8658815622329712, correct: False
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9087024927139282, max_false: 0.8014950752258301, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8821543455123901, max_false: 0.8284156918525696, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9000223278999329, max_false: 0.9002893567085266, correct: False
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8726707696914673, max_false: 0.8713827133178711, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9315932393074036, max_false: 0.9300983548164368, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9350257515907288, max_false: 0.906524658203125, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9203411936759949, max_false: 0.9213849306106567, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8278060555458069, max_false: 0.7994185090065002, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9327881336212158, max_false: 0.8772661089897156, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8919795155525208, max_false: 0.8555794358253479, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9099423885345459, max_false: 0.8442599177360535, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8218626379966736, max_false: 0.8786545991897583, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9523325562477112, max_false: 0.8758609294891357, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9333865642547607, max_false: 0.8626985549926758, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9653765559196472, max_false: 0.9549758434295654, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8599690198898315, max_false: 0.8759409189224243, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9281108975410461, max_false: 0.8912437558174133, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8667784333229065, max_false: 0.8734284043312073, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9022139310836792, max_false: 0.8422992825508118, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8243502974510193, max_false: 0.7560302019119263, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9197207689285278, max_false: 0.8893724083900452, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.869485080242157, max_false: 0.8237329721450806, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9365734457969666, max_false: 0.914190411567688, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8801608681678772, max_false: 0.8812984228134155, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9535924792289734, max_false: 0.944736659526825, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8352231383323669, max_false: 0.826416552066803, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9531970024108887, max_false: 0.9041969776153564, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9143744707107544, max_false: 0.880127489566803, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9009976387023926, max_false: 0.909491240978241, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9194452166557312, max_false: 0.902131974697113, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9339076280593872, max_false: 0.8894307613372803, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7941973209381104, max_false: 0.7889345288276672, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9436854124069214, max_false: 0.8592477440834045, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8791072964668274, max_false: 0.7880800366401672, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9359609484672546, max_false: 0.9187113046646118, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8990368843078613, max_false: 0.9097415208816528, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8857077956199646, max_false: 0.8337841033935547, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.7331118583679199, max_false: 0.7045152187347412, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9297120571136475, max_false: 0.9003060460090637, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9131476283073425, max_false: 0.9159340858459473, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9352612495422363, max_false: 0.8844025731086731, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9028732180595398, max_false: 0.888918399810791, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9470062255859375, max_false: 0.8696961998939514, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9132591485977173, max_false: 0.8726581931114197, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9302605986595154, max_false: 0.8792526721954346, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.885169267654419, max_false: 0.879016637802124, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9537606239318848, max_false: 0.9270317554473877, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9136053919792175, max_false: 0.9050593376159668, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9082342386245728, max_false: 0.8428966999053955, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8539384603500366, max_false: 0.8129968643188477, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9356011748313904, max_false: 0.8889521956443787, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.811856210231781, max_false: 0.8016071319580078, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8643363118171692, max_false: 0.8806477785110474, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8729075789451599, max_false: 0.8566283583641052, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9472780227661133, max_false: 0.921733558177948, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.895247757434845, max_false: 0.8619838953018188, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9509872198104858, max_false: 0.9174531698226929, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8385164141654968, max_false: 0.8264449834823608, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9233865141868591, max_false: 0.9205116629600525, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8914045095443726, max_false: 0.888891339302063, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9478538632392883, max_false: 0.9060461521148682, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9191901683807373, max_false: 0.9185022711753845, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9560297727584839, max_false: 0.9355608224868774, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9349852800369263, max_false: 0.9311123490333557, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9185307025909424, max_false: 0.8408620357513428, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.909334659576416, max_false: 0.8455850481987, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9063899517059326, max_false: 0.9136577248573303, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8535643219947815, max_false: 0.8800305724143982, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9105567336082458, max_false: 0.8908957242965698, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8129921555519104, max_false: 0.8410645127296448, correct: False
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9436144828796387, max_false: 0.924464225769043, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7926541566848755, max_false: 0.7539352178573608, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9155610203742981, max_false: 0.8556060791015625, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9054218530654907, max_false: 0.8683468699455261, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9325239062309265, max_false: 0.9163579940795898, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.902611255645752, max_false: 0.9126881957054138, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9335564374923706, max_false: 0.9214868545532227, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8775814175605774, max_false: 0.8743746280670166, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9344529509544373, max_false: 0.9036092162132263, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8670424818992615, max_false: 0.844002366065979, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9287580251693726, max_false: 0.9091410636901855, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8836935758590698, max_false: 0.863982617855072, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9572189450263977, max_false: 0.8798478245735168, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9188692569732666, max_false: 0.8457561731338501, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9457987546920776, max_false: 0.9290046095848083, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9240219593048096, max_false: 0.8732873201370239, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8958709836006165, max_false: 0.8738874197006226, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9126635193824768, max_false: 0.8793259859085083, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9277067184448242, max_false: 0.9116867780685425, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8966057896614075, max_false: 0.8839564323425293, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9060043096542358, max_false: 0.9031761884689331, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8227107524871826, max_false: 0.8282498717308044, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9507532119750977, max_false: 0.919556200504303, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9034172296524048, max_false: 0.892551600933075, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9647952914237976, max_false: 0.9295731782913208, correct: True
---------------------------------------
Summary for Split 'test':
 results: {'correct': {'cat_1': 210, 'cat_2': 235}, 'incorrect': {'cat_1': 40, 'cat_2': 15}}
 accuracy: 89.00%

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