----- test -----
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7461565136909485, max_false: 0.127879336476326, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.6969140768051147, max_false: 0.16368331015110016, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.8893997669219971, max_false: 0.015080399811267853, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6045688390731812, max_false: -0.08000736683607101, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7527440786361694, max_false: -0.10529257357120514, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6289562582969666, max_false: -0.09854820370674133, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.789924681186676, max_false: 0.18108893930912018, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6440086364746094, max_false: -0.11988066136837006, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8467134237289429, max_false: -0.09131477028131485, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.4264619052410126, max_false: 0.18674294650554657, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7263105511665344, max_false: -0.15196070075035095, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.24447685480117798, max_false: 0.4087696969509125, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8225926160812378, max_false: -0.14771129190921783, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6896278262138367, max_false: -0.14698700606822968, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.902030348777771, max_false: -0.187898650765419, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7119224667549133, max_false: -0.042799513787031174, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.49599963426589966, max_false: 0.2579478621482849, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8095933794975281, max_false: -0.2231426239013672, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7321281433105469, max_false: -0.1394665688276291, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6991478800773621, max_false: -0.07983825355768204, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.42357203364372253, max_false: 0.06519205868244171, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.5390729308128357, max_false: -0.079371377825737, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7929602265357971, max_false: 0.12571600079536438, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7082586884498596, max_false: -0.11003722995519638, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7269599437713623, max_false: 0.09452062100172043, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.4316983222961426, max_false: 0.08461087197065353, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.912659227848053, max_false: -0.13080157339572906, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.730645477771759, max_false: -0.08501896262168884, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8247420191764832, max_false: -0.02965778298676014, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.7555244565010071, max_false: -0.11941979825496674, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.29732292890548706, max_false: 0.19874969124794006, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.5651935338973999, max_false: 0.11153353005647659, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.944169282913208, max_false: -0.20209293067455292, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6282302141189575, max_false: -0.13435418903827667, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.25968924164772034, max_false: 0.23340515792369843, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6389461755752563, max_false: 0.15020836889743805, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6759042739868164, max_false: 0.03596247732639313, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6281143426895142, max_false: -0.09078372269868851, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.849888026714325, max_false: -0.0131235271692276, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.6413359045982361, max_false: 0.05308438092470169, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8801194429397583, max_false: -0.07410351186990738, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5114517211914062, max_false: -0.07312564551830292, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.880555272102356, max_false: 0.09334183484315872, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.564184844493866, max_false: 0.05524766817688942, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.91021728515625, max_false: -0.165244460105896, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.6157421469688416, max_false: -0.153593972325325, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.93808513879776, max_false: -0.1451798379421234, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.6852366328239441, max_false: -0.1394570916891098, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.893741250038147, max_false: -0.30582332611083984, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6515008807182312, max_false: -0.02287171222269535, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8167572021484375, max_false: -0.04853954166173935, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.5500842928886414, max_false: -0.08611688762903214, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.4600100517272949, max_false: 0.31636711955070496, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.569152295589447, max_false: -0.05366376042366028, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9326417446136475, max_false: -0.1936318576335907, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7283979654312134, max_false: -0.09316578507423401, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.3757344186306, max_false: 0.330305814743042, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7384153604507446, max_false: -0.07593300193548203, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.86476731300354, max_false: -0.17482270300388336, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7729519009590149, max_false: -0.2109394520521164, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8891505002975464, max_false: -0.18511340022087097, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.7007655501365662, max_false: -0.12390410155057907, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.4332042336463928, max_false: 0.35650187730789185, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.5540797710418701, max_false: -0.04899996519088745, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9009423851966858, max_false: -0.1617697775363922, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.6704509854316711, max_false: -0.16918346285820007, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7366204261779785, max_false: 0.021321063861250877, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.5368831753730774, max_false: -0.0007664700970053673, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9242271780967712, max_false: -0.15103714168071747, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.700880229473114, max_false: -0.1636282056570053, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.4817684590816498, max_false: 0.11037632822990417, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8231862187385559, max_false: -0.02650868520140648, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8740434050559998, max_false: -0.12283696234226227, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7683736681938171, max_false: -0.06348896026611328, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.7112923264503479, max_false: 0.2508867084980011, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7597308158874512, max_false: 0.3432583212852478, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9219616651535034, max_false: 0.0061864228919148445, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.670406699180603, max_false: -0.23857329785823822, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8607156276702881, max_false: -0.018275881186127663, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5922295451164246, max_false: -0.10677759349346161, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9335170984268188, max_false: -0.1897364854812622, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6212316751480103, max_false: -0.2632231116294861, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8174440264701843, max_false: -0.1305471956729889, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.510844349861145, max_false: 0.2696121335029602, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.915220320224762, max_false: -0.217960387468338, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.5860117673873901, max_false: -0.0217731986194849, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8659824132919312, max_false: -0.08610603958368301, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6607198119163513, max_false: 0.18122994899749756, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6959306001663208, max_false: 0.015896182507276535, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6600046157836914, max_false: 0.02664017304778099, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.17764946818351746, max_false: 0.48127979040145874, correct: False
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.054013535380363464, max_false: -0.03536909073591232, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8891115188598633, max_false: -0.20149847865104675, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5975920557975769, max_false: -0.12040681391954422, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8116961717605591, max_false: -0.026064442470669746, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.17378677427768707, max_false: 0.6556517481803894, correct: False
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8912625908851624, max_false: -0.19493409991264343, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.40962913632392883, max_false: 0.2076941430568695, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9377639889717102, max_false: -0.08434230834245682, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6405141353607178, max_false: 0.015546491369605064, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9356417655944824, max_false: -0.08631131052970886, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7353893518447876, max_false: -0.1283874809741974, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8559455275535583, max_false: 0.23844262957572937, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.6608530879020691, max_false: 0.025921329855918884, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6571282148361206, max_false: 0.25521811842918396, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.5519540905952454, max_false: 0.04085731506347656, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8632199764251709, max_false: -0.20826026797294617, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.7376503944396973, max_false: -0.13401669263839722, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8370428085327148, max_false: -0.017010726034641266, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6792156100273132, max_false: 0.00030048590269871056, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8416922688484192, max_false: -0.10096815973520279, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6453312635421753, max_false: -0.04449324682354927, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6079345941543579, max_false: 0.20592918992042542, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7007851600646973, max_false: -0.042627330869436264, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.6553884744644165, max_false: 0.02128913439810276, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.6977125406265259, max_false: -0.10397815704345703, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8977592587471008, max_false: -0.07952168583869934, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6022802591323853, max_false: 0.047684427350759506, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.888012170791626, max_false: -0.15736274421215057, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.22469277679920197, max_false: 0.5917995572090149, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8944082856178284, max_false: -0.11736655235290527, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7321610450744629, max_false: -0.10689087957143784, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.762717068195343, max_false: -0.0540890097618103, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.575904130935669, max_false: -0.06436130404472351, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.620636522769928, max_false: -0.11235986649990082, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.48710769414901733, max_false: 0.1247333511710167, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8855140209197998, max_false: 0.06492471694946289, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6981005072593689, max_false: -0.2536141574382782, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7395930886268616, max_false: 0.4710376560688019, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7543107867240906, max_false: 0.17761008441448212, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.7851100564002991, max_false: 0.04020818695425987, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.6396111845970154, max_false: -0.07849479466676712, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.945978581905365, max_false: -0.14043627679347992, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7845683097839355, max_false: -0.007660129107534885, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8863522410392761, max_false: 0.018290963023900986, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.7517898082733154, max_false: -0.011915267445147038, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9123860001564026, max_false: -0.03942086175084114, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.2875268757343292, max_false: 0.5910974740982056, correct: False
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.6892828345298767, max_false: -0.03295622766017914, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6838122010231018, max_false: -0.16101379692554474, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.905234694480896, max_false: -0.21390222012996674, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6575364470481873, max_false: -0.1774800717830658, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5868304371833801, max_false: 0.11158157140016556, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8084755539894104, max_false: -0.014867683872580528, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8331369757652283, max_false: -0.15484678745269775, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.642572283744812, max_false: -0.16022862493991852, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8559011816978455, max_false: -0.18853121995925903, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7366012334823608, max_false: -0.28418925404548645, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5042128562927246, max_false: 0.3325391411781311, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6500967144966125, max_false: 0.08920489996671677, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8198807835578918, max_false: 0.1996801793575287, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.621717631816864, max_false: -0.000536711944732815, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.5184086561203003, max_false: 0.34072667360305786, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5756811499595642, max_false: -0.10443484783172607, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7491821646690369, max_false: 0.07956899702548981, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.317929208278656, max_false: 0.4755374491214752, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8720663785934448, max_false: -0.1828821748495102, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7062562108039856, max_false: -0.11818481236696243, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8891651034355164, max_false: -0.16821660101413727, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5538804531097412, max_false: -0.09222971647977829, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9033088684082031, max_false: -0.22097353637218475, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6361334919929504, max_false: -0.19970263540744781, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8381702303886414, max_false: 0.07144247740507126, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.606162965297699, max_false: 0.1527942270040512, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.4008457660675049, max_false: 0.49341505765914917, correct: False
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5368609428405762, max_false: 0.44529256224632263, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9087579250335693, max_false: -0.17347288131713867, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6073076128959656, max_false: -0.06961885094642639, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8308448791503906, max_false: 0.06078948453068733, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.30847129225730896, max_false: 0.6036322712898254, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.49788686633110046, max_false: 0.004376396536827087, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5257288217544556, max_false: 0.21342973411083221, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.650094747543335, max_false: -0.02494724839925766, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6806166172027588, max_false: -0.18960537016391754, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.824962317943573, max_false: -0.12458673119544983, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7587146759033203, max_false: -0.058830857276916504, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.4371105134487152, max_false: 0.4044898450374603, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7087774872779846, max_false: -0.02620924636721611, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5878043174743652, max_false: 0.3845919370651245, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6131644248962402, max_false: 0.06568729132413864, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7621926069259644, max_false: -0.0014783815713599324, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.195997416973114, max_false: 0.5251355171203613, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6156094074249268, max_false: 0.2759702801704407, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7615687251091003, max_false: -0.03457147628068924, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.716564416885376, max_false: -0.15028251707553864, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.07844600826501846, max_false: -0.05881110578775406, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.9269002079963684, max_false: -0.09171324223279953, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.65522700548172, max_false: 0.0020269029773771763, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.905022382736206, max_false: -0.2762393057346344, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5834765434265137, max_false: -0.16438306868076324, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6228320002555847, max_false: 0.41416510939598083, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7948588728904724, max_false: -0.06765169650316238, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8443398475646973, max_false: -0.006453883368521929, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5962643623352051, max_false: -0.13153338432312012, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9036458134651184, max_false: -0.10273278504610062, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6387585401535034, max_false: -0.10476838052272797, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.5805342793464661, max_false: 0.31973910331726074, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.22893045842647552, max_false: 0.573642373085022, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.3740518093109131, max_false: 0.4775348901748657, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6466139554977417, max_false: -0.050994355231523514, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5788111090660095, max_false: 0.0503973513841629, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7417778372764587, max_false: -0.06378505378961563, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7543666362762451, max_false: -0.08763332664966583, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7333385944366455, max_false: -0.18103718757629395, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8686953783035278, max_false: 0.08252786099910736, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8379334211349487, max_false: -0.048447977751493454, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9055604934692383, max_false: -0.18501180410385132, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.6885464787483215, max_false: -0.15318624675273895, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8533298969268799, max_false: 0.04943934082984924, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6755669713020325, max_false: -0.1326906532049179, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.892665684223175, max_false: -0.03987862542271614, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5095242857933044, max_false: -0.1234651580452919, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7618513703346252, max_false: 0.005599521566182375, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.6387962698936462, max_false: 0.22940140962600708, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8817543387413025, max_false: -0.1020030751824379, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7338699698448181, max_false: -0.05203922837972641, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7476921677589417, max_false: -0.20721010863780975, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6217254400253296, max_false: -0.1799761950969696, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8929254412651062, max_false: -0.14855492115020752, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.7183741331100464, max_false: -0.11279579997062683, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.901585578918457, max_false: -0.2616925835609436, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.6758086681365967, max_false: -0.20872890949249268, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5546000003814697, max_false: 0.2513590157032013, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.6155209541320801, max_false: -0.08857424557209015, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.7086798548698425, max_false: 0.15079927444458008, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7775275111198425, max_false: -0.01977933943271637, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.889559268951416, max_false: 0.38097891211509705, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6075480580329895, max_false: -0.06069433316588402, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.878589391708374, max_false: -0.1054302453994751, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8187122941017151, max_false: -0.09223858267068863, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.89229816198349, max_false: -0.18205484747886658, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6774802803993225, max_false: -0.0874657928943634, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9369593262672424, max_false: -0.0031791995279490948, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7200901508331299, max_false: -0.18088659644126892, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7619761824607849, max_false: 0.014866444282233715, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.4450939893722534, max_false: 0.32224974036216736, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7848038077354431, max_false: -0.06826998293399811, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7915902733802795, max_false: -0.1607070416212082, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.8853659629821777, max_false: -0.16135790944099426, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.693079948425293, max_false: -0.09529054164886475, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9384167194366455, max_false: -0.19350534677505493, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.726584792137146, max_false: -0.1543704867362976, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8206143379211426, max_false: -0.18306535482406616, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.60933518409729, max_false: -0.1038760244846344, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7319145798683167, max_false: 0.14479118585586548, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.5325539112091064, max_false: 0.27463701367378235, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9351046085357666, max_false: -0.054989635944366455, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7454327940940857, max_false: -0.13518179953098297, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8990525603294373, max_false: -0.11307991296052933, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7418586015701294, max_false: -0.07274309545755386, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8682655096054077, max_false: 0.03532944247126579, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.5991107821464539, max_false: -0.0561695396900177, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8319745659828186, max_false: 0.04505350813269615, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7251313924789429, max_false: -0.21062040328979492, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8228352069854736, max_false: 0.07569658756256104, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.632386326789856, max_false: -0.09114217013120651, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.5010465979576111, max_false: 0.30461952090263367, correct: True
in: torch.Size([111, 5120]), out: torch.Size([5120]), max_true: 0.7665427327156067, max_false: 0.03439582511782646, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.8376156687736511, max_false: -0.13474446535110474, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6551063060760498, max_false: -0.17729437351226807, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6683505177497864, max_false: 0.08330460637807846, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.759495735168457, max_false: -0.00270733586512506, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9111414551734924, max_false: -0.3028757870197296, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5574280619621277, max_false: -0.15166732668876648, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8184040784835815, max_false: -0.12896601855754852, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7658608555793762, max_false: -0.13984183967113495, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9339033961296082, max_false: -0.11062957346439362, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8307870030403137, max_false: -0.06192627549171448, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5338361263275146, max_false: 0.2671184837818146, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.4652062654495239, max_false: 0.14736749231815338, correct: True
in: torch.Size([194, 5120]), out: torch.Size([5120]), max_true: 0.8409405946731567, max_false: -0.18133360147476196, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7158786654472351, max_false: -0.04048684239387512, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8855726718902588, max_false: -0.14474613964557648, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.5005670785903931, max_false: 0.10599129647016525, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.841545820236206, max_false: -0.20479051768779755, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7181376218795776, max_false: -0.04929886385798454, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.49726322293281555, max_false: 0.10819835215806961, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6571328043937683, max_false: -0.013337837532162666, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8930901885032654, max_false: -0.1633901596069336, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.5953616499900818, max_false: -0.01805083267390728, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9353246092796326, max_false: -0.20937177538871765, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7216983437538147, max_false: -0.15653841197490692, correct: True
in: torch.Size([74, 5120]), out: torch.Size([5120]), max_true: -0.21805746853351593, max_false: 0.6309468150138855, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5657856464385986, max_false: 0.16633237898349762, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8759409785270691, max_false: -0.08409738540649414, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7650793790817261, max_false: -0.11101050674915314, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6473587155342102, max_false: -0.0622495599091053, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7749779224395752, max_false: -0.15090899169445038, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6725701689720154, max_false: 0.24406857788562775, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6768273711204529, max_false: 0.021171830594539642, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8350610733032227, max_false: -0.26732078194618225, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.43541577458381653, max_false: -0.06413687765598297, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8836333155632019, max_false: -0.17548562586307526, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7637865543365479, max_false: -0.11956579983234406, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7492720484733582, max_false: -0.03682902082800865, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7292098999023438, max_false: -0.12095848470926285, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7743252515792847, max_false: 0.027016574516892433, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7986618280410767, max_false: -0.0064185312949121, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.450107216835022, max_false: 0.2522323429584503, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6417343616485596, max_false: 0.010845297947525978, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9418880939483643, max_false: -0.07638063281774521, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8149849772453308, max_false: -0.06047830358147621, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7897883653640747, max_false: -0.08455521613359451, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6794490814208984, max_false: -0.017276739701628685, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8659254908561707, max_false: -0.06533541530370712, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.656333327293396, max_false: 0.018844975158572197, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8810139894485474, max_false: -0.24233494699001312, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7515767216682434, max_false: -0.21115770936012268, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6067424416542053, max_false: -0.10500234365463257, correct: True
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.6930229067802429, max_false: -0.11460097134113312, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7172768115997314, max_false: -0.11586697399616241, correct: True
in: torch.Size([89, 5120]), out: torch.Size([5120]), max_true: 0.6953561305999756, max_false: -0.19541124999523163, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8771823048591614, max_false: -0.10314371436834335, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7193090915679932, max_false: -0.20811620354652405, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.570371150970459, max_false: 0.1405247151851654, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7443183660507202, max_false: -0.19353248178958893, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.855609655380249, max_false: -0.04971975088119507, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7020899653434753, max_false: -0.025590043514966965, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8553173542022705, max_false: -0.09996329247951508, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7930546402931213, max_false: -0.23149454593658447, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8380607962608337, max_false: 0.022158261388540268, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5564033389091492, max_false: 0.008834069594740868, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.6658523082733154, max_false: -0.06944695860147476, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6832670569419861, max_false: -0.06218906119465828, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8295210599899292, max_false: -0.052399106323719025, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.46527764201164246, max_false: 0.18206031620502472, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9209337830543518, max_false: -0.260414183139801, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7265681028366089, max_false: -0.25076824426651, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.9229556918144226, max_false: -0.016698410734534264, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7184259295463562, max_false: -0.03647632524371147, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9294230937957764, max_false: -0.08738771080970764, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.751925528049469, max_false: -0.17698222398757935, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8714170455932617, max_false: -0.08506757766008377, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6662031412124634, max_false: -0.15811459720134735, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8424069285392761, max_false: -0.041773874312639236, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7258408069610596, max_false: -0.26423949003219604, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8901875019073486, max_false: 0.04569798335433006, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.6732261180877686, max_false: 0.35673364996910095, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7850900292396545, max_false: -0.001120546949096024, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5947426557540894, max_false: -0.15710186958312988, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9447656869888306, max_false: -0.18810681998729706, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6822106838226318, max_false: -0.1358756422996521, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9150322079658508, max_false: -0.22626571357250214, correct: True
in: torch.Size([104, 5120]), out: torch.Size([5120]), max_true: 0.7812557816505432, max_false: -0.2129727452993393, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8760783076286316, max_false: -0.020173244178295135, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7074137926101685, max_false: -0.13768313825130463, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7656684517860413, max_false: 0.02164224348962307, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.6168051958084106, max_false: 0.09241122752428055, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8859004378318787, max_false: -0.22937189042568207, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7628002166748047, max_false: -0.17030750215053558, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6888464689254761, max_false: -0.1709764003753662, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6325132846832275, max_false: -0.07506269961595535, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9021124839782715, max_false: -0.1745557337999344, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.782477617263794, max_false: -0.21615442633628845, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6210880279541016, max_false: 0.006245302967727184, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.38608062267303467, max_false: -0.017736021429300308, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7648497819900513, max_false: 0.05412355810403824, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.17604364454746246, max_false: 0.8107384443283081, correct: False
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7034683227539062, max_false: -0.00999204721301794, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9069163799285889, max_false: -0.17994898557662964, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.68177729845047, max_false: -0.1330132931470871, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8680150508880615, max_false: 0.060981862246990204, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8032512664794922, max_false: -0.11698352545499802, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8767465353012085, max_false: -0.11492139846086502, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6966716647148132, max_false: -0.016948776319622993, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6273950934410095, max_false: 0.01976957730948925, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7032454609870911, max_false: -0.0429956279695034, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8471174240112305, max_false: -0.09153129160404205, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7924370169639587, max_false: -0.07429080456495285, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8953795433044434, max_false: -0.03887343779206276, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7482736110687256, max_false: -0.10604386031627655, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.789287269115448, max_false: -0.027803605422377586, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6121506094932556, max_false: -0.011013687588274479, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9181768298149109, max_false: -0.07060530036687851, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6057733297348022, max_false: 0.0005767904222011566, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9196872115135193, max_false: -0.2520950138568878, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5328654050827026, max_false: -0.1204291433095932, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9236581921577454, max_false: -0.21394117176532745, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.6976410746574402, max_false: -0.12884828448295593, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6779049634933472, max_false: -0.12218616157770157, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6687862873077393, max_false: -0.1271084100008011, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5667403936386108, max_false: 0.14078612625598907, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6174020171165466, max_false: -0.07097253948450089, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8591674566268921, max_false: -0.07712336629629135, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.7390435934066772, max_false: -0.14830687642097473, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7267731428146362, max_false: 0.10187043994665146, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6182627081871033, max_false: -0.08809899538755417, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9281937479972839, max_false: -0.0934838056564331, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5436576008796692, max_false: 0.03256552666425705, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5466633439064026, max_false: 0.2667514383792877, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7313404083251953, max_false: -0.149262472987175, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9287353157997131, max_false: -0.13353557884693146, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7738873362541199, max_false: -0.27705517411231995, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7993221282958984, max_false: -0.00707068108022213, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5499598979949951, max_false: -0.0713779404759407, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8796824812889099, max_false: -0.12871506810188293, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8107101321220398, max_false: -0.1811380237340927, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7566927671432495, max_false: 0.1964082568883896, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5435444116592407, max_false: 0.09195312857627869, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9102230668067932, max_false: -0.0330592542886734, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.3704269528388977, max_false: 0.7928650975227356, correct: False
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8087688088417053, max_false: 0.02377975918352604, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.4476810693740845, max_false: 0.38638484477996826, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7390233874320984, max_false: 0.08367972075939178, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6728516221046448, max_false: 0.01756076142191887, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.6457680463790894, max_false: -0.015592367388308048, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7003461122512817, max_false: -0.10133492946624756, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.33999693393707275, max_false: 0.34180089831352234, correct: False
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.6362953782081604, max_false: 0.0005325102247297764, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9099134206771851, max_false: 0.112380750477314, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6247105002403259, max_false: 0.03949202597141266, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8724328875541687, max_false: -0.07271380722522736, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6027765274047852, max_false: -0.12683069705963135, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.6540626287460327, max_false: 0.21494945883750916, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.6304921507835388, max_false: -0.21610330045223236, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7338911294937134, max_false: -0.023194238543510437, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.40752217173576355, max_false: 0.35172489285469055, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.8787474632263184, max_false: -0.030072666704654694, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7936378717422485, max_false: -0.05825338512659073, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9388153553009033, max_false: 0.007796327583491802, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.5502300262451172, max_false: 0.06668630242347717, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8206051588058472, max_false: -0.1646849364042282, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5004664063453674, max_false: -0.13004115223884583, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.85353684425354, max_false: 0.10847780853509903, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.6264840364456177, max_false: -0.009783047251403332, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8184775114059448, max_false: -0.025543367490172386, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5450299382209778, max_false: 0.22415949404239655, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9122363924980164, max_false: -0.01800742745399475, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.548610270023346, max_false: 0.04926728084683418, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8369629979133606, max_false: -0.2363416701555252, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6526432633399963, max_false: -0.19028212130069733, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.9105769991874695, max_false: -0.05056306719779968, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7480068802833557, max_false: -0.09495393931865692, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5059899687767029, max_false: 0.14689764380455017, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8005145192146301, max_false: -0.025223590433597565, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.5944742560386658, max_false: -0.002977019874379039, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5397133231163025, max_false: 0.021327940747141838, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.9199872016906738, max_false: -0.12483339011669159, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7342870235443115, max_false: -0.04195316135883331, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.9111362099647522, max_false: -0.17442330718040466, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.46174901723861694, max_false: 0.2291164994239807, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6410971283912659, max_false: -0.18922390043735504, correct: True
in: torch.Size([195, 5120]), out: torch.Size([5120]), max_true: 0.6087881922721863, max_false: -0.1624549776315689, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.9018415212631226, max_false: -0.13953977823257446, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.5725167989730835, max_false: 0.00251106359064579, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.7955874800682068, max_false: -0.11511535942554474, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6618743538856506, max_false: 0.027791615575551987, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.8768311738967896, max_false: -0.003208315931260586, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6986455917358398, max_false: 0.04312954470515251, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.5467746257781982, max_false: 0.24928393959999084, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.6513144373893738, max_false: 0.08495878428220749, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8151569366455078, max_false: 0.11014614254236221, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7294357419013977, max_false: -0.01640506647527218, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.812241792678833, max_false: -0.27273261547088623, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6619938015937805, max_false: -0.20244784653186798, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.8972193598747253, max_false: -0.16505126655101776, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7235987782478333, max_false: -0.24301089346408844, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.049860160797834396, max_false: 0.4873374402523041, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.18822619318962097, max_false: 0.5162460207939148, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.8951554894447327, max_false: 0.025990508496761322, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7377588748931885, max_false: -0.06390215456485748, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.8141952157020569, max_false: 0.04877777770161629, correct: True
in: torch.Size([139, 5120]), out: torch.Size([5120]), max_true: 0.6461778283119202, max_false: 0.04096691310405731, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.5818502306938171, max_false: 0.33169710636138916, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.6923936009407043, max_false: -0.1529584527015686, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7853649258613586, max_false: -0.03196068853139877, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.45571741461753845, max_false: 0.26989656686782837, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.9411098957061768, max_false: -0.25516295433044434, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.15386086702346802, max_false: 0.6116135120391846, correct: False
in: torch.Size([179, 5120]), out: torch.Size([5120]), max_true: 0.6291844248771667, max_false: -0.044763341546058655, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7610260844230652, max_false: -0.15275020897388458, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6743044853210449, max_false: 0.14872491359710693, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.6787477731704712, max_false: 0.011291653849184513, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5693696141242981, max_false: 0.17615552246570587, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.5525326728820801, max_false: -0.04810873419046402, correct: True
in: torch.Size([181, 5120]), out: torch.Size([5120]), max_true: 0.9351420998573303, max_false: 0.0483897402882576, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.7688764929771423, max_false: 0.08014657348394394, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8171549439430237, max_false: -0.09491502493619919, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7697586417198181, max_false: -0.15093140304088593, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.799214243888855, max_false: 0.05941832438111305, correct: True
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.7039408087730408, max_false: -0.12854748964309692, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.766325056552887, max_false: 0.13996146619319916, correct: True
in: torch.Size([167, 5120]), out: torch.Size([5120]), max_true: 0.6309062838554382, max_false: -0.04460928216576576, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8391250371932983, max_false: 0.10356108099222183, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.61765056848526, max_false: 0.14474129676818848, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7947036027908325, max_false: 0.0804624930024147, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7177445292472839, max_false: -0.1020560935139656, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.9321866631507874, max_false: -0.14451414346694946, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7228348851203918, max_false: -0.06915812194347382, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.7420056462287903, max_false: -0.10089468955993652, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7946482300758362, max_false: -0.15992912650108337, correct: True
in: torch.Size([119, 5120]), out: torch.Size([5120]), max_true: 0.8012163639068604, max_false: -0.15260356664657593, correct: True
in: torch.Size([164, 5120]), out: torch.Size([5120]), max_true: 0.8167868852615356, max_false: -0.13133160769939423, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.7303798198699951, max_false: 0.11333201825618744, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.7533013820648193, max_false: -0.08646232634782791, correct: True
in: torch.Size([134, 5120]), out: torch.Size([5120]), max_true: 0.48790106177330017, max_false: 0.4749040901660919, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.3962947130203247, max_false: 0.42349234223365784, correct: False
in: torch.Size([149, 5120]), out: torch.Size([5120]), max_true: 0.844617486000061, max_false: 0.02544122189283371, correct: True
in: torch.Size([153, 5120]), out: torch.Size([5120]), max_true: 0.5685819983482361, max_false: 0.22905980050563812, correct: True
in: torch.Size([209, 5120]), out: torch.Size([5120]), max_true: 0.8721125721931458, max_false: -0.04103140905499458, correct: True
[481]