----- test -----
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8915137052536011, max_false: 0.86232590675354, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8604081869125366, max_false: 0.8216172456741333, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.9033788442611694, max_false: 0.8905373811721802, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8490254878997803, max_false: 0.8229840993881226, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9214379191398621, max_false: 0.8838850855827332, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9278825521469116, max_false: 0.8681537508964539, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9274998307228088, max_false: 0.9062461256980896, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8782060146331787, max_false: 0.910659909248352, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9057990312576294, max_false: 0.8860949277877808, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8823736310005188, max_false: 0.862747848033905, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8693176507949829, max_false: 0.8388646841049194, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8545848727226257, max_false: 0.8895575404167175, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9498229026794434, max_false: 0.8892215490341187, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8992667198181152, max_false: 0.8475226163864136, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9377404451370239, max_false: 0.8433849215507507, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9034834504127502, max_false: 0.8357282280921936, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9307802319526672, max_false: 0.9043598175048828, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8958393335342407, max_false: 0.8730696439743042, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9062691330909729, max_false: 0.861838161945343, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9145081043243408, max_false: 0.9151801466941833, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8838269710540771, max_false: 0.8773424625396729, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8270536065101624, max_false: 0.8291627168655396, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.943091094493866, max_false: 0.9341089129447937, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8368096351623535, max_false: 0.8392201662063599, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9277356266975403, max_false: 0.9132340550422668, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8176360130310059, max_false: 0.8219367265701294, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9493883848190308, max_false: 0.9230801463127136, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9337847232818604, max_false: 0.9080622792243958, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9258111715316772, max_false: 0.8932852149009705, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8817980289459229, max_false: 0.8822325468063354, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8797515034675598, max_false: 0.910106897354126, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.870929479598999, max_false: 0.8842774629592896, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9313502311706543, max_false: 0.8664148449897766, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8259415626525879, max_false: 0.7881644368171692, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.7251906991004944, max_false: 0.6312500834465027, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8753888010978699, max_false: 0.8670519590377808, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8712917566299438, max_false: 0.8243374228477478, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8582145571708679, max_false: 0.7986477613449097, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9600810408592224, max_false: 0.9446130990982056, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8940656781196594, max_false: 0.8870537281036377, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9439505338668823, max_false: 0.904252290725708, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8411110043525696, max_false: 0.8288400769233704, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9374173879623413, max_false: 0.9168784022331238, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8530650734901428, max_false: 0.8493531942367554, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9504584670066833, max_false: 0.8669525980949402, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8490409255027771, max_false: 0.8021017909049988, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9398523569107056, max_false: 0.8859492540359497, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8769412040710449, max_false: 0.8694530725479126, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9479533433914185, max_false: 0.8942095041275024, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8815155029296875, max_false: 0.8849602341651917, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9029057025909424, max_false: 0.8814677000045776, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7902908325195312, max_false: 0.7640437483787537, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9267107844352722, max_false: 0.9216538667678833, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.8597885966300964, max_false: 0.8441639542579651, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9621479511260986, max_false: 0.8510544300079346, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9138525128364563, max_false: 0.8057665824890137, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9017834663391113, max_false: 0.8733609914779663, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8754573464393616, max_false: 0.8728839159011841, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9297335147857666, max_false: 0.8623492121696472, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8959209322929382, max_false: 0.8617057800292969, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9349474906921387, max_false: 0.8624284267425537, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.8919609189033508, max_false: 0.8337658643722534, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9335931539535522, max_false: 0.937728226184845, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8518401384353638, max_false: 0.85066819190979, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9398123025894165, max_false: 0.9328116774559021, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.909959614276886, max_false: 0.9179733991622925, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9107440114021301, max_false: 0.8823491334915161, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.897504448890686, max_false: 0.8903778195381165, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9241796731948853, max_false: 0.8906500339508057, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8521113395690918, max_false: 0.828605592250824, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8785136938095093, max_false: 0.8360548615455627, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9522535800933838, max_false: 0.9035102128982544, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9229505062103271, max_false: 0.9035495519638062, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8962627649307251, max_false: 0.8550264239311218, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9320375919342041, max_false: 0.9299476146697998, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9252926111221313, max_false: 0.9325727224349976, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9374571442604065, max_false: 0.9360817074775696, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8756760954856873, max_false: 0.8410140872001648, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9223489761352539, max_false: 0.8396244645118713, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8115910291671753, max_false: 0.7221936583518982, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9475738406181335, max_false: 0.8817510604858398, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7988182902336121, max_false: 0.8013910055160522, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9315049648284912, max_false: 0.9070855379104614, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8932838439941406, max_false: 0.8968890309333801, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9140386581420898, max_false: 0.8126956224441528, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8333209156990051, max_false: 0.7971494197845459, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9263730049133301, max_false: 0.8263181447982788, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9335669875144958, max_false: 0.8270028233528137, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9100221991539001, max_false: 0.8900155425071716, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9008575081825256, max_false: 0.8494910597801208, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7900466918945312, max_false: 0.7638779878616333, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.17170163989067078, max_false: 0.15247325599193573, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9514167308807373, max_false: 0.8597749471664429, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9029248952865601, max_false: 0.8298646211624146, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9142591953277588, max_false: 0.9398699402809143, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9461459517478943, max_false: 0.9109053611755371, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9189168214797974, max_false: 0.8959167003631592, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9029492139816284, max_false: 0.8899275064468384, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.936678409576416, max_false: 0.933902382850647, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.914635181427002, max_false: 0.9001728296279907, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9376971125602722, max_false: 0.890028715133667, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.856855571269989, max_false: 0.8192271590232849, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9552395343780518, max_false: 0.7291600704193115, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8188274502754211, max_false: 0.6525292992591858, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8888259530067444, max_false: 0.8979328870773315, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9109269976615906, max_false: 0.9111897945404053, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9542785882949829, max_false: 0.8761340379714966, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.8933433294296265, max_false: 0.8553466200828552, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9345166087150574, max_false: 0.8881043195724487, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9054713845252991, max_false: 0.8665716052055359, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9495373368263245, max_false: 0.9083783626556396, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8442826271057129, max_false: 0.8358772397041321, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9092012047767639, max_false: 0.8920013308525085, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9355001449584961, max_false: 0.9085708856582642, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9109982252120972, max_false: 0.8631932139396667, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8670631051063538, max_false: 0.8250302076339722, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9327645897865295, max_false: 0.8155622482299805, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8927560448646545, max_false: 0.8497831225395203, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.9587069153785706, max_false: 0.9280335307121277, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9017163515090942, max_false: 0.8934163451194763, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9454101324081421, max_false: 0.9058617353439331, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8874801993370056, max_false: 0.8399142026901245, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9326385259628296, max_false: 0.8580341935157776, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8548672795295715, max_false: 0.8285397291183472, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8620454668998718, max_false: 0.8273710012435913, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8474968075752258, max_false: 0.8108915090560913, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9336094856262207, max_false: 0.9074733853340149, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8812616467475891, max_false: 0.8762207627296448, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9430942535400391, max_false: 0.928624153137207, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.925531804561615, max_false: 0.9225234985351562, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8847163319587708, max_false: 0.8492946028709412, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.7995651364326477, max_false: 0.7973559498786926, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9459662437438965, max_false: 0.8636946678161621, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9198351502418518, max_false: 0.8792905211448669, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9420962333679199, max_false: 0.8518804311752319, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8243204355239868, max_false: 0.7803429961204529, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9291779398918152, max_false: 0.874566912651062, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.890177309513092, max_false: 0.908250093460083, correct: False
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9228291511535645, max_false: 0.8930426836013794, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8851661086082458, max_false: 0.8574177026748657, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9334928393363953, max_false: 0.9239682555198669, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8574281334877014, max_false: 0.8512938618659973, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9062026143074036, max_false: 0.8980444073677063, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9332008361816406, max_false: 0.8671666383743286, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9184563159942627, max_false: 0.9057019352912903, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8634946942329407, max_false: 0.8542076349258423, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9495061635971069, max_false: 0.9237356185913086, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8722022771835327, max_false: 0.8464075326919556, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8885350227355957, max_false: 0.8947727680206299, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9232485890388489, max_false: 0.8712726831436157, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9485608339309692, max_false: 0.9472708702087402, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8981157541275024, max_false: 0.8726552724838257, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9077627658843994, max_false: 0.8798145055770874, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7591333985328674, max_false: 0.7618892788887024, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9198522567749023, max_false: 0.9051975011825562, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9195907115936279, max_false: 0.8955559730529785, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.936623752117157, max_false: 0.9119774103164673, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9109314680099487, max_false: 0.8831944465637207, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.927283763885498, max_false: 0.8426449298858643, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8169398307800293, max_false: 0.7417643666267395, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9311776161193848, max_false: 0.8761966824531555, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8829730153083801, max_false: 0.8360693454742432, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9422258734703064, max_false: 0.9023312330245972, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9096817374229431, max_false: 0.8814632892608643, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9228548407554626, max_false: 0.9350150227546692, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8691047430038452, max_false: 0.8524597883224487, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.937128484249115, max_false: 0.9205772876739502, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.874483048915863, max_false: 0.8930115699768066, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9552697539329529, max_false: 0.9348028898239136, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9249517321586609, max_false: 0.9272655844688416, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.882623016834259, max_false: 0.8165883421897888, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8940459489822388, max_false: 0.8599690198898315, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8643935918807983, max_false: 0.8637725710868835, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8458749055862427, max_false: 0.7961380481719971, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9223109483718872, max_false: 0.9121266603469849, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9199193716049194, max_false: 0.9191425442695618, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9215736985206604, max_false: 0.9166461229324341, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8655860424041748, max_false: 0.854511559009552, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8795280456542969, max_false: 0.862399160861969, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9043168425559998, max_false: 0.8662248253822327, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9229708909988403, max_false: 0.9039433598518372, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8403850793838501, max_false: 0.8826712369918823, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.929490327835083, max_false: 0.9057836532592773, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8975785374641418, max_false: 0.8684365153312683, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8996176719665527, max_false: 0.8021191358566284, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.17412897944450378, max_false: 0.15799199044704437, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9528986811637878, max_false: 0.9150221347808838, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8608565926551819, max_false: 0.8470808267593384, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.944075345993042, max_false: 0.882438600063324, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8763938546180725, max_false: 0.8228990435600281, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9564555883407593, max_false: 0.9464660286903381, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9032958149909973, max_false: 0.8933299779891968, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9159011840820312, max_false: 0.8762478232383728, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.702864944934845, max_false: 0.6973441243171692, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9478263854980469, max_false: 0.8654723763465881, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8370944857597351, max_false: 0.7984360456466675, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9296555519104004, max_false: 0.9406819939613342, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9294706583023071, max_false: 0.9309876561164856, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9040282964706421, max_false: 0.9066405892372131, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8617937564849854, max_false: 0.8325373530387878, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9032930135726929, max_false: 0.9145791530609131, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8939627408981323, max_false: 0.8845371603965759, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8878785371780396, max_false: 0.8882999420166016, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8182169198989868, max_false: 0.8265494108200073, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.946158766746521, max_false: 0.8878328204154968, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9237895011901855, max_false: 0.8642832040786743, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9471058249473572, max_false: 0.9227076768875122, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8507062196731567, max_false: 0.8452174067497253, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9573658108711243, max_false: 0.902630627155304, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9011585116386414, max_false: 0.8464328050613403, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9493818879127502, max_false: 0.9306780695915222, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7561632990837097, max_false: 0.7786949872970581, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9384632110595703, max_false: 0.9066954851150513, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9271169304847717, max_false: 0.9221136569976807, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.925334632396698, max_false: 0.8861550092697144, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9204426407814026, max_false: 0.8980222344398499, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9389253258705139, max_false: 0.8717247247695923, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8729192614555359, max_false: 0.8407390117645264, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9043490290641785, max_false: 0.8350297212600708, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8939329385757446, max_false: 0.8350661396980286, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9243974685668945, max_false: 0.8868575096130371, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.9005457758903503, max_false: 0.8643308877944946, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9151625037193298, max_false: 0.8769113421440125, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.8442559242248535, max_false: 0.8119860291481018, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9287845492362976, max_false: 0.9193488359451294, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8711149096488953, max_false: 0.8327687978744507, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9536687731742859, max_false: 0.9487830996513367, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8864669799804688, max_false: 0.8839871287345886, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9479692578315735, max_false: 0.8748186230659485, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9327306151390076, max_false: 0.871803343296051, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.921446681022644, max_false: 0.8178817629814148, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8693640828132629, max_false: 0.7972133755683899, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9516273736953735, max_false: 0.9161002039909363, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8778774738311768, max_false: 0.8403034210205078, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9317120313644409, max_false: 0.9072856307029724, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.856065571308136, max_false: 0.891843855381012, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9406705498695374, max_false: 0.9211255311965942, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9328957200050354, max_false: 0.9022899270057678, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.9318504929542542, max_false: 0.9103643894195557, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8805293440818787, max_false: 0.8195832967758179, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.958762526512146, max_false: 0.8934429287910461, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9430835843086243, max_false: 0.8723628520965576, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9471073746681213, max_false: 0.9179696440696716, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.8733035326004028, max_false: 0.8605946898460388, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9264734983444214, max_false: 0.9031757116317749, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8901746869087219, max_false: 0.8952562808990479, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9608594179153442, max_false: 0.9298017024993896, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9033325910568237, max_false: 0.8524251580238342, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9422314763069153, max_false: 0.9104979634284973, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8990594744682312, max_false: 0.8810950517654419, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9188550710678101, max_false: 0.8854063153266907, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8886969685554504, max_false: 0.8625251650810242, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9376677870750427, max_false: 0.9467998743057251, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8613482117652893, max_false: 0.8553730249404907, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9414695501327515, max_false: 0.9105756282806396, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.820164144039154, max_false: 0.7883107662200928, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8990302085876465, max_false: 0.8783758282661438, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.9217392206192017, max_false: 0.8407947421073914, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9389878511428833, max_false: 0.909327507019043, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8856184482574463, max_false: 0.8638678789138794, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9283841848373413, max_false: 0.8906188011169434, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.913450300693512, max_false: 0.8832173943519592, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9026119709014893, max_false: 0.875089168548584, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8421097993850708, max_false: 0.8576515913009644, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9437176585197449, max_false: 0.9059115052223206, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9072335958480835, max_false: 0.8555238246917725, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9548859000205994, max_false: 0.9152902960777283, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9087943434715271, max_false: 0.8692467212677002, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8405372500419617, max_false: 0.7693886160850525, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.8661220073699951, max_false: 0.834191620349884, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.9542378187179565, max_false: 0.9341962337493896, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9363500475883484, max_false: 0.9041377902030945, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9258896112442017, max_false: 0.873589813709259, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8747813701629639, max_false: 0.8903072476387024, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9148032665252686, max_false: 0.9004682302474976, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.909660279750824, max_false: 0.8545133471488953, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9138059020042419, max_false: 0.8435019254684448, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9019361734390259, max_false: 0.8764384388923645, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9484058022499084, max_false: 0.9250723123550415, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.7721630930900574, max_false: 0.7865238785743713, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9449417591094971, max_false: 0.9219095706939697, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8827277421951294, max_false: 0.8493449687957764, correct: True
in: torch.Size([74, 5120]), out: torch.Size([5120]), max_true: 0.821677029132843, max_false: 0.7949259281158447, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9179539084434509, max_false: 0.8830826282501221, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9404590725898743, max_false: 0.8620719909667969, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8873504400253296, max_false: 0.8056742548942566, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9137511849403381, max_false: 0.8529629111289978, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9061626195907593, max_false: 0.8452219367027283, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9339834451675415, max_false: 0.8907751441001892, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8963319063186646, max_false: 0.8671244382858276, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8935205340385437, max_false: 0.8616974353790283, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8324753046035767, max_false: 0.8599390983581543, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9266896843910217, max_false: 0.9024441242218018, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9171661734580994, max_false: 0.9107333421707153, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9163256883621216, max_false: 0.909360408782959, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8495455384254456, max_false: 0.8503538966178894, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9444715976715088, max_false: 0.9334307909011841, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.913348913192749, max_false: 0.8752990365028381, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8998442888259888, max_false: 0.8926422595977783, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.883952260017395, max_false: 0.8881253600120544, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9469215273857117, max_false: 0.8733094930648804, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9135075807571411, max_false: 0.8567348718643188, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9306437969207764, max_false: 0.860008955001831, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9216804504394531, max_false: 0.895572304725647, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9305741786956787, max_false: 0.8822896480560303, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9017394185066223, max_false: 0.87213534116745, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9460285305976868, max_false: 0.9083357453346252, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9163529276847839, max_false: 0.9022818803787231, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8606652021408081, max_false: 0.8228253126144409, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8957725763320923, max_false: 0.8322066068649292, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9137493968009949, max_false: 0.8717494606971741, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.8659054040908813, max_false: 0.8406949043273926, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9088438749313354, max_false: 0.9128254055976868, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8504422307014465, max_false: 0.8462817668914795, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9508625268936157, max_false: 0.932378888130188, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9206554293632507, max_false: 0.9083207249641418, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9482663869857788, max_false: 0.8965429067611694, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.885446310043335, max_false: 0.861578643321991, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9491224884986877, max_false: 0.8855077028274536, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.879357099533081, max_false: 0.8130265474319458, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9262599349021912, max_false: 0.8735323548316956, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8419461250305176, max_false: 0.8059951663017273, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9345801472663879, max_false: 0.8748153448104858, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9152558445930481, max_false: 0.8658210635185242, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9241954684257507, max_false: 0.8961635231971741, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9261919856071472, max_false: 0.8712883591651917, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9577078819274902, max_false: 0.909717321395874, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8983793258666992, max_false: 0.8756598234176636, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9459990859031677, max_false: 0.9408465623855591, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9107595086097717, max_false: 0.9087167978286743, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9046653509140015, max_false: 0.9108147025108337, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8450201153755188, max_false: 0.82009357213974, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9431887269020081, max_false: 0.9006410241127014, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8631730079650879, max_false: 0.8521429896354675, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9418760538101196, max_false: 0.9245116114616394, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8589403033256531, max_false: 0.8402523994445801, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.941236138343811, max_false: 0.9111281037330627, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9228529334068298, max_false: 0.9041550159454346, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9444965720176697, max_false: 0.9193790555000305, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8428323864936829, max_false: 0.8291329741477966, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9457190036773682, max_false: 0.8901790380477905, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8519096374511719, max_false: 0.8115941286087036, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.933268129825592, max_false: 0.9234765768051147, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.8930572867393494, max_false: 0.8688522577285767, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9269000291824341, max_false: 0.912742018699646, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8996782898902893, max_false: 0.8803775310516357, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9378653168678284, max_false: 0.9279325008392334, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8812297582626343, max_false: 0.8698475360870361, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9217320084571838, max_false: 0.820140540599823, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9241283535957336, max_false: 0.8710904717445374, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8748072981834412, max_false: 0.8432528376579285, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.86773681640625, max_false: 0.8038750290870667, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9363448023796082, max_false: 0.9239063262939453, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8805403709411621, max_false: 0.8530338406562805, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8915870785713196, max_false: 0.8580763339996338, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.756916880607605, max_false: 0.7472501993179321, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9209267497062683, max_false: 0.9153228402137756, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9268770813941956, max_false: 0.8852983713150024, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9181711077690125, max_false: 0.8740982413291931, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9505491256713867, max_false: 0.854989230632782, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8977687954902649, max_false: 0.8523728847503662, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9368471503257751, max_false: 0.8943658471107483, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9285362958908081, max_false: 0.8621296882629395, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9209393858909607, max_false: 0.901076078414917, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9406633973121643, max_false: 0.9196705222129822, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.917551577091217, max_false: 0.9014092683792114, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9014251232147217, max_false: 0.8829900026321411, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9259177446365356, max_false: 0.8914436101913452, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9370882511138916, max_false: 0.9040473103523254, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9367970824241638, max_false: 0.8957445621490479, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8715174198150635, max_false: 0.8895384669303894, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9160978198051453, max_false: 0.8348513841629028, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9043580293655396, max_false: 0.8127071261405945, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9267264604568481, max_false: 0.8992506265640259, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9297115206718445, max_false: 0.8636550903320312, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9100338220596313, max_false: 0.8222057819366455, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8591886162757874, max_false: 0.7788718342781067, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9267103672027588, max_false: 0.8919424414634705, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9023375511169434, max_false: 0.8862677216529846, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8840563297271729, max_false: 0.7784478664398193, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8482511639595032, max_false: 0.7831352949142456, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8697555661201477, max_false: 0.8517836332321167, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8568276166915894, max_false: 0.8401677012443542, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.890350341796875, max_false: 0.8002483248710632, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8675122261047363, max_false: 0.8047483563423157, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9512761235237122, max_false: 0.9227005243301392, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9053288102149963, max_false: 0.9364507794380188, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9401183724403381, max_false: 0.8490301966667175, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8539611101150513, max_false: 0.8177499175071716, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9049943685531616, max_false: 0.8758364319801331, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9342839121818542, max_false: 0.8834791779518127, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9340634942054749, max_false: 0.8951960802078247, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8808772563934326, max_false: 0.8337677121162415, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.900202214717865, max_false: 0.792827844619751, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7917242050170898, max_false: 0.7884023189544678, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.943314254283905, max_false: 0.9181998372077942, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9009920954704285, max_false: 0.8781540989875793, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9011173248291016, max_false: 0.881323516368866, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8807424306869507, max_false: 0.8825063109397888, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9385050535202026, max_false: 0.9030002951622009, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9222872853279114, max_false: 0.8917210102081299, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9385650753974915, max_false: 0.9175666570663452, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8241010308265686, max_false: 0.825325608253479, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9160254001617432, max_false: 0.8981019854545593, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8410316705703735, max_false: 0.8496454358100891, correct: False
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8955362439155579, max_false: 0.7942304611206055, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8775670528411865, max_false: 0.8299421072006226, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9008693695068359, max_false: 0.9047723412513733, correct: False
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8633456826210022, max_false: 0.8692415356636047, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9306200742721558, max_false: 0.9425470232963562, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9297168254852295, max_false: 0.9101067781448364, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9240689277648926, max_false: 0.9189336895942688, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8191659450531006, max_false: 0.7918486595153809, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9288774132728577, max_false: 0.8831544518470764, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8720040321350098, max_false: 0.8463205099105835, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9023685455322266, max_false: 0.8334112167358398, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8354933261871338, max_false: 0.8864360451698303, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9507100582122803, max_false: 0.8609588742256165, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.92464679479599, max_false: 0.8515454530715942, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.962449312210083, max_false: 0.9529503583908081, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.8610042333602905, max_false: 0.8783562183380127, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9292047023773193, max_false: 0.8863524198532104, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8479619026184082, max_false: 0.8613578677177429, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9032158255577087, max_false: 0.8457878232002258, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.7974345684051514, max_false: 0.7456997036933899, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9141974449157715, max_false: 0.8872314691543579, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8671830892562866, max_false: 0.8212445378303528, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9369645714759827, max_false: 0.9215151071548462, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8710220456123352, max_false: 0.8686898946762085, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9436413645744324, max_false: 0.9415116310119629, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8466432690620422, max_false: 0.8253715634346008, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9511515498161316, max_false: 0.9032905101776123, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9140656590461731, max_false: 0.879752516746521, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9122365713119507, max_false: 0.9091801047325134, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9167009592056274, max_false: 0.8982653617858887, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9180999994277954, max_false: 0.8847305774688721, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7660918235778809, max_false: 0.7738763093948364, correct: False
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9308437705039978, max_false: 0.8541300296783447, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8746100068092346, max_false: 0.8023800849914551, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9353047609329224, max_false: 0.9232189655303955, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9038003087043762, max_false: 0.908907949924469, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8741743564605713, max_false: 0.8177192211151123, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.7312602400779724, max_false: 0.7111101746559143, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9316835403442383, max_false: 0.9005293846130371, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9030032157897949, max_false: 0.910489559173584, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9270563125610352, max_false: 0.8791255950927734, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8959544897079468, max_false: 0.8716736435890198, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9324018359184265, max_false: 0.8664751648902893, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9050138592720032, max_false: 0.8672613501548767, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9321801066398621, max_false: 0.874225914478302, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8729405999183655, max_false: 0.8701512813568115, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9477344751358032, max_false: 0.9261729717254639, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.907051146030426, max_false: 0.8994279503822327, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8977625966072083, max_false: 0.8334227800369263, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8475303053855896, max_false: 0.8046867847442627, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9305621385574341, max_false: 0.8702071905136108, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8130253553390503, max_false: 0.791585385799408, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.863269567489624, max_false: 0.8876357674598694, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8757812976837158, max_false: 0.869391143321991, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9449756145477295, max_false: 0.91617751121521, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8901907801628113, max_false: 0.8488081693649292, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.9424036741256714, max_false: 0.9029889106750488, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8135027289390564, max_false: 0.7828658223152161, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9071599245071411, max_false: 0.9161933064460754, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8828015923500061, max_false: 0.8797867894172668, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9418246746063232, max_false: 0.9065691232681274, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9171395301818848, max_false: 0.9119526147842407, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9542251825332642, max_false: 0.9304667711257935, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9303624033927917, max_false: 0.9282991290092468, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9136667251586914, max_false: 0.8369055986404419, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8940990567207336, max_false: 0.858905553817749, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8920934200286865, max_false: 0.9059039354324341, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8486658334732056, max_false: 0.8634225726127625, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9005144834518433, max_false: 0.8857550621032715, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.813768744468689, max_false: 0.8341853618621826, correct: False
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9477379322052002, max_false: 0.9219642281532288, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.806454598903656, max_false: 0.7720831632614136, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9081776142120361, max_false: 0.8446935415267944, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8950597643852234, max_false: 0.8639655709266663, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9346950650215149, max_false: 0.9191829562187195, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8925606608390808, max_false: 0.898674726486206, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.929115891456604, max_false: 0.9232965111732483, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8675171732902527, max_false: 0.8682037591934204, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9332860112190247, max_false: 0.9018881320953369, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8677064776420593, max_false: 0.8394460678100586, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9183526635169983, max_false: 0.8975337743759155, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8862244486808777, max_false: 0.8660662770271301, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9568502902984619, max_false: 0.8836768269538879, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9209054112434387, max_false: 0.8521324396133423, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9440709352493286, max_false: 0.9300242066383362, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9167154431343079, max_false: 0.8796221017837524, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8833457827568054, max_false: 0.87272709608078, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9115710854530334, max_false: 0.8596546053886414, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9207690954208374, max_false: 0.9059479832649231, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9013341665267944, max_false: 0.8808823823928833, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8962734341621399, max_false: 0.8919878602027893, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8119290471076965, max_false: 0.8058459758758545, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9503510594367981, max_false: 0.9176527857780457, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9018332958221436, max_false: 0.8942378163337708, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9634808301925659, max_false: 0.938534140586853, correct: True
[436]