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.6960917711257935, max_false: 0.6570682525634766, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9593033194541931, max_false: 0.912073016166687, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.9707244038581848, max_false: 0.9276971817016602, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8395781517028809, max_false: 0.8133272528648376, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9515523910522461, max_false: 0.9640321731567383, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9488505721092224, max_false: 0.9441136121749878, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9531932473182678, max_false: 0.9362257719039917, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9524941444396973, max_false: 0.9410972595214844, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9102041125297546, max_false: 0.8455458283424377, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8466147780418396, max_false: 0.8904399871826172, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8749635219573975, max_false: 0.8748109340667725, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.939903736114502, max_false: 0.9583330154418945, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9749830365180969, max_false: 0.9537590146064758, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9215205907821655, max_false: 0.9034044146537781, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9784013628959656, max_false: 0.9614207148551941, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9538472890853882, max_false: 0.9322516918182373, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9693053364753723, max_false: 0.9585328698158264, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9631958603858948, max_false: 0.9456647038459778, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9779112339019775, max_false: 0.9493653774261475, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9407913088798523, max_false: 0.9197971224784851, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9771825075149536, max_false: 0.9771662950515747, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.903215765953064, max_false: 0.9125515222549438, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9740240573883057, max_false: 0.9625967741012573, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7851048707962036, max_false: 0.7907479405403137, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9651310443878174, max_false: 0.9767581820487976, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9468076229095459, max_false: 0.9360915422439575, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.976183295249939, max_false: 0.9830306172370911, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9774685502052307, max_false: 0.9689297676086426, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.959446370601654, max_false: 0.9446631669998169, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9227951169013977, max_false: 0.9293712377548218, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9527404308319092, max_false: 0.9701351523399353, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9553254842758179, max_false: 0.963792622089386, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8683577179908752, max_false: 0.7917114496231079, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9409687519073486, max_false: 0.9118443727493286, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9022255539894104, max_false: 0.8967835903167725, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9678763151168823, max_false: 0.9633746147155762, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.948992133140564, max_false: 0.9457383155822754, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9398633241653442, max_false: 0.9237500429153442, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.972055196762085, max_false: 0.9839545488357544, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9756256937980652, max_false: 0.96433424949646, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9683277010917664, max_false: 0.9764775633811951, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9638742208480835, max_false: 0.9549933671951294, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.956149697303772, max_false: 0.9445928335189819, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9411271214485168, max_false: 0.9396262764930725, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9481698274612427, max_false: 0.9472233057022095, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9598167538642883, max_false: 0.9481003880500793, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9816042184829712, max_false: 0.9680672883987427, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9599823951721191, max_false: 0.9497985243797302, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.98513263463974, max_false: 0.9558599591255188, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9479439854621887, max_false: 0.9562027454376221, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9558060169219971, max_false: 0.9429331421852112, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9250426888465881, max_false: 0.9118455648422241, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9628412127494812, max_false: 0.9502488374710083, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9303187131881714, max_false: 0.9226024150848389, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9823281764984131, max_false: 0.9587160348892212, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9669215083122253, max_false: 0.9369304776191711, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9764412641525269, max_false: 0.9688597917556763, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9555568099021912, max_false: 0.9609021544456482, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9616001844406128, max_false: 0.9389362931251526, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9559213519096375, max_false: 0.9432002305984497, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9134432673454285, max_false: 0.9180641174316406, correct: False
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.9208223819732666, max_false: 0.946486234664917, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9648253917694092, max_false: 0.9718065857887268, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9407140612602234, max_false: 0.9405705332756042, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9696987867355347, max_false: 0.9740946888923645, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9274397492408752, max_false: 0.9433453679084778, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9507701992988586, max_false: 0.9554120898246765, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9584659934043884, max_false: 0.9641184210777283, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9790809750556946, max_false: 0.9589244723320007, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.96688312292099, max_false: 0.951930046081543, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9578758478164673, max_false: 0.9591909646987915, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9582131505012512, max_false: 0.9405365586280823, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9715522527694702, max_false: 0.9540094137191772, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9344598054885864, max_false: 0.9322932958602905, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.988311767578125, max_false: 0.9853855967521667, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9653816819190979, max_false: 0.9657324552536011, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.985702633857727, max_false: 0.97959965467453, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8771487474441528, max_false: 0.8523669838905334, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9807984828948975, max_false: 0.9526674151420593, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9285997748374939, max_false: 0.9085758924484253, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9030788540840149, max_false: 0.8181862831115723, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9592939615249634, max_false: 0.9056378602981567, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.979469895362854, max_false: 0.9512194991111755, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9683525562286377, max_false: 0.9511274099349976, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.963338315486908, max_false: 0.9481019973754883, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9367666840553284, max_false: 0.9258930683135986, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7645977139472961, max_false: 0.7398289442062378, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9624307751655579, max_false: 0.9428209066390991, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9632930159568787, max_false: 0.9602141380310059, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9473714828491211, max_false: 0.9328673481941223, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9483181834220886, max_false: 0.6226195693016052, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.47274255752563477, max_false: 0.4547591507434845, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9781976342201233, max_false: 0.9577794075012207, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9500682950019836, max_false: 0.9155893921852112, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.834404468536377, max_false: 0.8528082370758057, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9732344150543213, max_false: 0.9382181167602539, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9662570953369141, max_false: 0.9656747579574585, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9592973589897156, max_false: 0.9604877829551697, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9529282450675964, max_false: 0.9429009556770325, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.963934063911438, max_false: 0.9662361741065979, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.787386417388916, max_false: 0.7565380930900574, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.937874436378479, max_false: 0.9292095899581909, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7094956636428833, max_false: 0.47538286447525024, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9566404819488525, max_false: 0.8549787998199463, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9679156541824341, max_false: 0.9748407602310181, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9455782771110535, max_false: 0.9427894949913025, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9659184813499451, max_false: 0.9739122986793518, correct: False
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9649382829666138, max_false: 0.9637941122055054, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9669081568717957, max_false: 0.9650737047195435, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9661064147949219, max_false: 0.9440485239028931, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9835702776908875, max_false: 0.9714247584342957, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.837093710899353, max_false: 0.8624594807624817, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8300389051437378, max_false: 0.8463588953018188, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9624719023704529, max_false: 0.948499321937561, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9557113647460938, max_false: 0.9452778697013855, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9381709694862366, max_false: 0.9281485676765442, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9673988819122314, max_false: 0.9305258989334106, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9615556001663208, max_false: 0.9389026165008545, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.9850696921348572, max_false: 0.9854838252067566, correct: False
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9855799674987793, max_false: 0.9737347960472107, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9466217756271362, max_false: 0.8902078866958618, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9469200968742371, max_false: 0.9457319378852844, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9604395031929016, max_false: 0.9445591568946838, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9510862827301025, max_false: 0.9328477382659912, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.959132194519043, max_false: 0.9458293914794922, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9591086506843567, max_false: 0.9388351440429688, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6153589487075806, max_false: 0.47992753982543945, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9595792293548584, max_false: 0.9153738617897034, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9784578084945679, max_false: 0.9674886465072632, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.962127685546875, max_false: 0.9581357836723328, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9505290389060974, max_false: 0.9454852342605591, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9549409747123718, max_false: 0.9373741149902344, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9794960021972656, max_false: 0.9600567817687988, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9379316568374634, max_false: 0.9242560267448425, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9852629899978638, max_false: 0.9643524885177612, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.972109854221344, max_false: 0.9650862216949463, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9843416213989258, max_false: 0.9668600559234619, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9582198858261108, max_false: 0.9725748300552368, correct: False
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9844908714294434, max_false: 0.9717191457748413, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9560791254043579, max_false: 0.9533658623695374, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9843103885650635, max_false: 0.9683037996292114, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9484263062477112, max_false: 0.9476685523986816, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.971998929977417, max_false: 0.9385260343551636, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9521176218986511, max_false: 0.8783559799194336, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9643401503562927, max_false: 0.9688855409622192, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.954928457736969, max_false: 0.9577913284301758, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9765611886978149, max_false: 0.9613484144210815, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.925959587097168, max_false: 0.9304571747779846, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9609021544456482, max_false: 0.9516943693161011, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9380322098731995, max_false: 0.910307765007019, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9554910063743591, max_false: 0.9512493014335632, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9364274144172668, max_false: 0.9283264875411987, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9508306980133057, max_false: 0.9627541303634644, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9248748421669006, max_false: 0.9360657930374146, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9759228229522705, max_false: 0.9672751426696777, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9758011698722839, max_false: 0.9692168831825256, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9878111481666565, max_false: 0.9721721410751343, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.975385308265686, max_false: 0.972460925579071, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.96108078956604, max_false: 0.9298741221427917, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.93272465467453, max_false: 0.9418061971664429, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9621860980987549, max_false: 0.929324746131897, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9562920928001404, max_false: 0.9416971802711487, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9674732089042664, max_false: 0.9645735025405884, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9418085813522339, max_false: 0.9341212511062622, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9747153520584106, max_false: 0.9831025004386902, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9514320492744446, max_false: 0.9220611453056335, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9748402833938599, max_false: 0.9712170362472534, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9679142832756042, max_false: 0.9736800789833069, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9863731265068054, max_false: 0.9538419246673584, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.974096953868866, max_false: 0.9512112736701965, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.95408695936203, max_false: 0.9438053965568542, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9654321670532227, max_false: 0.9397481083869934, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9496691226959229, max_false: 0.9491510987281799, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9416337013244629, max_false: 0.9538657665252686, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9618799090385437, max_false: 0.9579660892486572, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9617958664894104, max_false: 0.9650095701217651, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9716238379478455, max_false: 0.9649718403816223, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9187552332878113, max_false: 0.9103805422782898, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9425318837165833, max_false: 0.9266891479492188, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.963381290435791, max_false: 0.9448420405387878, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.981036901473999, max_false: 0.9743625521659851, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9106766581535339, max_false: 0.9277417063713074, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8872178196907043, max_false: 0.8837655782699585, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.979476809501648, max_false: 0.9727609753608704, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9450452923774719, max_false: 0.6892896890640259, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.4755290150642395, max_false: 0.4472440183162689, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.986844539642334, max_false: 0.9621351361274719, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.951543390750885, max_false: 0.95965176820755, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9730351567268372, max_false: 0.94553142786026, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9523160457611084, max_false: 0.9257615208625793, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9751235842704773, max_false: 0.9759957790374756, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9643396139144897, max_false: 0.9606637358665466, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9601958394050598, max_false: 0.9399213790893555, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6097397804260254, max_false: 0.6215568780899048, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.985896110534668, max_false: 0.9676977396011353, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9139593243598938, max_false: 0.9052391052246094, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9597314596176147, max_false: 0.9579544067382812, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8689233660697937, max_false: 0.89304518699646, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9681346416473389, max_false: 0.9648644328117371, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9171476364135742, max_false: 0.9103698134422302, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9752812385559082, max_false: 0.9744040966033936, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9059067964553833, max_false: 0.9147950410842896, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9687212109565735, max_false: 0.973518967628479, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8088534474372864, max_false: 0.8163998126983643, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8840493559837341, max_false: 0.8456651568412781, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9589962959289551, max_false: 0.9223315715789795, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9851917624473572, max_false: 0.9632315635681152, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.9599248170852661, max_false: 0.9297205209732056, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9816280603408813, max_false: 0.9720383286476135, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9030569791793823, max_false: 0.8777366280555725, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9897001385688782, max_false: 0.9754130244255066, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9016193151473999, max_false: 0.9068735837936401, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9755899310112, max_false: 0.9710497856140137, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9656755924224854, max_false: 0.970650315284729, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9808632135391235, max_false: 0.9553498029708862, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9038952589035034, max_false: 0.8857757449150085, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9879913330078125, max_false: 0.9695814847946167, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9412078857421875, max_false: 0.9280125498771667, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9739946126937866, max_false: 0.9517620801925659, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8839593529701233, max_false: 0.8379352688789368, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9634503126144409, max_false: 0.9519577026367188, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.9716857671737671, max_false: 0.9070905447006226, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9669853448867798, max_false: 0.9401929974555969, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.9046260118484497, max_false: 0.8977464437484741, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9781972169876099, max_false: 0.965634822845459, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9531478881835938, max_false: 0.9228208661079407, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9880999326705933, max_false: 0.9815564155578613, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8241976499557495, max_false: 0.820537805557251, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8956852555274963, max_false: 0.820105791091919, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9808610081672668, max_false: 0.9452852010726929, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9875470995903015, max_false: 0.9537118673324585, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9630445837974548, max_false: 0.9374043345451355, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9800758361816406, max_false: 0.978396475315094, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9566727876663208, max_false: 0.9459639191627502, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9590055346488953, max_false: 0.9736495614051819, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7993475198745728, max_false: 0.8846126794815063, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9806427359580994, max_false: 0.9449435472488403, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9457687139511108, max_false: 0.9237149953842163, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9749282598495483, max_false: 0.9628674983978271, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9575719833374023, max_false: 0.954808235168457, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9797751307487488, max_false: 0.9597205519676208, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9043521285057068, max_false: 0.8828458189964294, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9838308095932007, max_false: 0.9676191210746765, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.9632156491279602, max_false: 0.9464713931083679, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9626283645629883, max_false: 0.9531780481338501, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6557095646858215, max_false: 0.6844773888587952, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9892128109931946, max_false: 0.9694486856460571, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8887941837310791, max_false: 0.8606703281402588, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.976521909236908, max_false: 0.9517552852630615, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9649931192398071, max_false: 0.9181734919548035, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9751976132392883, max_false: 0.9307047724723816, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9243619441986084, max_false: 0.9099198579788208, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.984218180179596, max_false: 0.9791379570960999, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.919525146484375, max_false: 0.8990259170532227, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9421383738517761, max_false: 0.9466269016265869, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8781246542930603, max_false: 0.8648035526275635, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.973953366279602, max_false: 0.964034378528595, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.977514386177063, max_false: 0.9531728029251099, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.97568279504776, max_false: 0.9722336530685425, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9598956108093262, max_false: 0.9569966197013855, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7679704427719116, max_false: 0.7864862680435181, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9443486332893372, max_false: 0.9341133236885071, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.979854941368103, max_false: 0.9682061076164246, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9480664730072021, max_false: 0.9439565539360046, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.972061812877655, max_false: 0.9445095062255859, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9725434184074402, max_false: 0.9443895816802979, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9950416088104248, max_false: 0.9839453101158142, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9762722253799438, max_false: 0.9722634553909302, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9116775989532471, max_false: 0.8770747780799866, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.9209027886390686, max_false: 0.8960403203964233, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.9703453779220581, max_false: 0.9648689031600952, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9497485160827637, max_false: 0.9590395092964172, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9510135054588318, max_false: 0.9377819299697876, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9260062575340271, max_false: 0.9245125651359558, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9831377863883972, max_false: 0.9462147951126099, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9802178144454956, max_false: 0.9684064984321594, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9789611101150513, max_false: 0.9532893896102905, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9714279770851135, max_false: 0.9449443817138672, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9796788692474365, max_false: 0.9736875891685486, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9103524088859558, max_false: 0.9156150221824646, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.972410261631012, max_false: 0.9666345715522766, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9243822693824768, max_false: 0.9280990958213806, correct: False
in: torch.Size([74, 5120]), out: torch.Size([5120]), max_true: 0.6810554265975952, max_false: 0.7292052507400513, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9545239210128784, max_false: 0.9552667140960693, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8127622008323669, max_false: 0.7766133546829224, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9653371572494507, max_false: 0.9160254001617432, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.958686888217926, max_false: 0.9310138821601868, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9564909934997559, max_false: 0.9675100445747375, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9609055519104004, max_false: 0.9260523915290833, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9548973441123962, max_false: 0.9624142646789551, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9367870688438416, max_false: 0.941317617893219, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9515504837036133, max_false: 0.9627216458320618, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9838995337486267, max_false: 0.9722105264663696, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9748405814170837, max_false: 0.9675085544586182, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9774828553199768, max_false: 0.9788867235183716, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.961633563041687, max_false: 0.9598656892776489, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8799559473991394, max_false: 0.8451322317123413, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9723238348960876, max_false: 0.9624788761138916, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9323492646217346, max_false: 0.9004107713699341, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9511337876319885, max_false: 0.9527377486228943, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9697464108467102, max_false: 0.9245643615722656, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9541141986846924, max_false: 0.9315036535263062, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.979485034942627, max_false: 0.9571124315261841, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9702285528182983, max_false: 0.9476338624954224, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9749740362167358, max_false: 0.9609663486480713, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9741222858428955, max_false: 0.9623614549636841, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9818622469902039, max_false: 0.9670699834823608, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9680647850036621, max_false: 0.9660941362380981, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9621294736862183, max_false: 0.9566364288330078, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9662038087844849, max_false: 0.9564035534858704, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9808247685432434, max_false: 0.9759976863861084, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.933937668800354, max_false: 0.9367644786834717, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9802650809288025, max_false: 0.9778497815132141, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9615625739097595, max_false: 0.9586547017097473, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9799811244010925, max_false: 0.9536330103874207, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9172778129577637, max_false: 0.8799373507499695, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9698914289474487, max_false: 0.9505079388618469, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.943122148513794, max_false: 0.9311656355857849, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8892588019371033, max_false: 0.8391444087028503, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9455403089523315, max_false: 0.9163762331008911, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7943034172058105, max_false: 0.7727311849594116, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8922264575958252, max_false: 0.8979734778404236, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9730710387229919, max_false: 0.9513060450553894, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9722660183906555, max_false: 0.9548036456108093, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9666765332221985, max_false: 0.9564359784126282, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9720878005027771, max_false: 0.962719202041626, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9846633672714233, max_false: 0.9502259492874146, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9585180282592773, max_false: 0.9563831090927124, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9881148934364319, max_false: 0.9755195379257202, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9521763324737549, max_false: 0.9712162613868713, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9761334657669067, max_false: 0.9803728461265564, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9576103091239929, max_false: 0.9498690962791443, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9847914576530457, max_false: 0.9669308066368103, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9478696584701538, max_false: 0.9283937811851501, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9865050911903381, max_false: 0.9767959713935852, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.946894109249115, max_false: 0.941149115562439, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9313049912452698, max_false: 0.9672910571098328, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9700573086738586, max_false: 0.9487316012382507, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9793967008590698, max_false: 0.9587501287460327, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.918714165687561, max_false: 0.917137086391449, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9791918396949768, max_false: 0.9370701909065247, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8320566415786743, max_false: 0.7889443635940552, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9709639549255371, max_false: 0.9624313712120056, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9721289873123169, max_false: 0.9576284885406494, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9630388617515564, max_false: 0.9563828706741333, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9757633805274963, max_false: 0.9715262055397034, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9694654941558838, max_false: 0.9709059000015259, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9009841084480286, max_false: 0.9244335293769836, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8141242265701294, max_false: 0.8186912536621094, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8354639410972595, max_false: 0.863180935382843, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9551376700401306, max_false: 0.9481658339500427, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9519739747047424, max_false: 0.9388117790222168, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9828859567642212, max_false: 0.9666851162910461, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9424501657485962, max_false: 0.9408197402954102, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9602282047271729, max_false: 0.9431202411651611, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9445124268531799, max_false: 0.9290446043014526, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.885766327381134, max_false: 0.8300194144248962, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9653128385543823, max_false: 0.8911029696464539, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9563800096511841, max_false: 0.940238893032074, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9824091196060181, max_false: 0.953856885433197, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9492805004119873, max_false: 0.9265552163124084, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9766643047332764, max_false: 0.9576321244239807, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9565116763114929, max_false: 0.9467920064926147, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9555596113204956, max_false: 0.9666875600814819, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.974315881729126, max_false: 0.9689567685127258, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9555498957633972, max_false: 0.9536730647087097, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9604651927947998, max_false: 0.960733950138092, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9801473617553711, max_false: 0.9602435231208801, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8652368187904358, max_false: 0.7846790552139282, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9750293493270874, max_false: 0.8891233205795288, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8331608176231384, max_false: 0.9368769526481628, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9588842988014221, max_false: 0.950456976890564, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9510759711265564, max_false: 0.948554515838623, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9522191882133484, max_false: 0.9499386548995972, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9752820134162903, max_false: 0.9463057518005371, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9628114104270935, max_false: 0.9454226493835449, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9529481530189514, max_false: 0.8563427925109863, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.953782320022583, max_false: 0.9509665966033936, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9614375829696655, max_false: 0.9657471776008606, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9708302617073059, max_false: 0.9366083145141602, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9665753245353699, max_false: 0.9508993625640869, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8303536176681519, max_false: 0.8395001888275146, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9584091305732727, max_false: 0.9463204145431519, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9189543724060059, max_false: 0.9568154811859131, correct: False
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9506168961524963, max_false: 0.9003112316131592, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9281219244003296, max_false: 0.926300048828125, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9583238363265991, max_false: 0.9547719359397888, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9718974828720093, max_false: 0.9470678567886353, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9570931792259216, max_false: 0.9278488159179688, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9027876257896423, max_false: 0.8842819929122925, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9657015800476074, max_false: 0.9423932433128357, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9636207818984985, max_false: 0.9431402087211609, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9313087463378906, max_false: 0.915959894657135, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8750357031822205, max_false: 0.7217666506767273, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9220043420791626, max_false: 0.8348979353904724, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9827432036399841, max_false: 0.970714807510376, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9498441219329834, max_false: 0.9539865851402283, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9471031427383423, max_false: 0.9527696371078491, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9638137817382812, max_false: 0.9620028138160706, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9722578525543213, max_false: 0.965019702911377, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9685713052749634, max_false: 0.9428927898406982, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9778928756713867, max_false: 0.9649665355682373, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9485342502593994, max_false: 0.9129306077957153, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9776724576950073, max_false: 0.9644739031791687, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8064649701118469, max_false: 0.8426207304000854, correct: False
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9711978435516357, max_false: 0.929800808429718, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9304710626602173, max_false: 0.9459090828895569, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9708696603775024, max_false: 0.9476228952407837, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.955061137676239, max_false: 0.9562405943870544, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.961004376411438, max_false: 0.9539192318916321, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.94304358959198, max_false: 0.9268984198570251, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9760873317718506, max_false: 0.960647702217102, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8929064273834229, max_false: 0.9516539573669434, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9813765287399292, max_false: 0.9687554240226746, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9667103290557861, max_false: 0.9615920782089233, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9814324378967285, max_false: 0.9645224213600159, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9423750638961792, max_false: 0.9563935399055481, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9756327867507935, max_false: 0.9693490862846375, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9619196653366089, max_false: 0.9427312016487122, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.985393762588501, max_false: 0.9849391579627991, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9150232672691345, max_false: 0.9197491407394409, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9810951352119446, max_false: 0.9664219617843628, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9477150440216064, max_false: 0.9486876130104065, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.934278666973114, max_false: 0.9328752756118774, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9146326780319214, max_false: 0.9295387268066406, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9408241510391235, max_false: 0.9388097524642944, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9575396776199341, max_false: 0.9554988145828247, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9890123009681702, max_false: 0.984410285949707, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9684106707572937, max_false: 0.9569120407104492, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.987184464931488, max_false: 0.9826102256774902, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9792271256446838, max_false: 0.9798113107681274, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9374890923500061, max_false: 0.8430723547935486, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9496809840202332, max_false: 0.9178564548492432, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9626769423484802, max_false: 0.9644069671630859, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8563666939735413, max_false: 0.8274917006492615, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9428215026855469, max_false: 0.8981289863586426, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8706546425819397, max_false: 0.8546812534332275, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.976387083530426, max_false: 0.9649455547332764, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9619361758232117, max_false: 0.9329587817192078, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9847813248634338, max_false: 0.9695143699645996, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9424185752868652, max_false: 0.9316074252128601, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9135082364082336, max_false: 0.9107089042663574, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.9624675512313843, max_false: 0.9485634565353394, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9595836997032166, max_false: 0.9435018301010132, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9662895202636719, max_false: 0.9697015881538391, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.974298357963562, max_false: 0.9584682583808899, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9723786115646362, max_false: 0.9579826593399048, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9730061292648315, max_false: 0.9664645195007324, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9267317652702332, max_false: 0.9263877868652344, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.968034029006958, max_false: 0.9565686583518982, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8184949159622192, max_false: 0.7874832153320312, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9775500297546387, max_false: 0.9588668942451477, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9627549648284912, max_false: 0.9701939821243286, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9715719819068909, max_false: 0.9543124437332153, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9424681067466736, max_false: 0.9293285608291626, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9079810380935669, max_false: 0.8866863250732422, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8698651790618896, max_false: 0.8733243942260742, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9720796942710876, max_false: 0.9671273231506348, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9511629343032837, max_false: 0.945037305355072, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9756900072097778, max_false: 0.9725422859191895, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7466710209846497, max_false: 0.7177761197090149, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9472753405570984, max_false: 0.9519965052604675, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9062910676002502, max_false: 0.8845052719116211, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.965873658657074, max_false: 0.9629738330841064, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.935906708240509, max_false: 0.9310396909713745, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9825502038002014, max_false: 0.9708267450332642, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9687167406082153, max_false: 0.9711329936981201, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.988278329372406, max_false: 0.9679864048957825, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.975540280342102, max_false: 0.9473401308059692, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9659520983695984, max_false: 0.9347192645072937, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9391326308250427, max_false: 0.9467571377754211, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9639087915420532, max_false: 0.9669789671897888, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9136010408401489, max_false: 0.9285315871238708, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9438372850418091, max_false: 0.9191007018089294, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8767585158348083, max_false: 0.8702912926673889, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9709804058074951, max_false: 0.9660388827323914, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9328900575637817, max_false: 0.9415134191513062, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9736148118972778, max_false: 0.926967978477478, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9183914065361023, max_false: 0.9365721940994263, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8174549341201782, max_false: 0.7832977175712585, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8372807502746582, max_false: 0.8826190233230591, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9758801460266113, max_false: 0.969829261302948, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9494870901107788, max_false: 0.960343062877655, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9023983478546143, max_false: 0.9404517412185669, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9100308418273926, max_false: 0.9220516681671143, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9452505111694336, max_false: 0.9511055946350098, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9568125009536743, max_false: 0.9365869164466858, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9802367091178894, max_false: 0.9410440921783447, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9613271355628967, max_false: 0.9339134693145752, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9863382577896118, max_false: 0.9808014631271362, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8993774652481079, max_false: 0.9080434441566467, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.934166669845581, max_false: 0.9336561560630798, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9366073608398438, max_false: 0.9203876852989197, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.955488383769989, max_false: 0.9602789878845215, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9526903033256531, max_false: 0.9145614504814148, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.6689915657043457, max_false: 0.5574869513511658, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9357192516326904, max_false: 0.8996846079826355, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9755403399467468, max_false: 0.9547066688537598, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9716910719871521, max_false: 0.9594482183456421, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.983901858329773, max_false: 0.9647226333618164, correct: True
---------------------------------------
Summary for Split 'test':
 results: {'correct': {'cat_1': 179, 'cat_2': 209}, 'incorrect': {'cat_1': 71, 'cat_2': 41}}
 accuracy: 77.60%

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