Precision: [tensor(1., device='cuda:0'), tensor(1., device='cuda:0'), tensor(1., device='cuda:0'), tensor(1., device='cuda:0'), tensor(0.9998, device='cuda:0'), tensor(1., device='cuda:0'), tensor(0.9998, device='cuda:0'), tensor(1., device='cuda:0'), tensor(0.9998, device='cuda:0'), tensor(0.9998, device='cuda:0')]

Output distance: [tensor(38781.6562, device='cuda:0'), tensor(38996.0117, device='cuda:0'), tensor(40191.2500, device='cuda:0'), tensor(39072.6562, device='cuda:0'), tensor(39412.5820, device='cuda:0'), tensor(39028.9219, device='cuda:0'), tensor(38820.1797, device='cuda:0'), tensor(39309.0703, device='cuda:0'), tensor(39175.0312, device='cuda:0'), tensor(39051.2539, device='cuda:0')]

Prediction loss: [tensor(37292.3672, device='cuda:0'), tensor(36996.6211, device='cuda:0'), tensor(40197.6602, device='cuda:0'), tensor(40425.5312, device='cuda:0'), tensor(41111.3125, device='cuda:0'), tensor(38856.1875, device='cuda:0'), tensor(39722.1992, device='cuda:0'), tensor(40686.7852, device='cuda:0'), tensor(37210.7188, device='cuda:0'), tensor(39186.5977, device='cuda:0')]

Others: [{'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}, {'num_positive': tensor(6000, device='cuda:0'), 'num_positive_true': tensor(18000, device='cuda:0')}]

Compressed training loss: [tensor(3465445., device='cuda:0'), tensor(3406959.7500, device='cuda:0'), tensor(3545076.2500, device='cuda:0'), tensor(3599069.2500, device='cuda:0'), tensor(3636651.7500, device='cuda:0'), tensor(3563454., device='cuda:0'), tensor(3617433.5000, device='cuda:0'), tensor(3628032., device='cuda:0'), tensor(3387049., device='cuda:0'), tensor(3607706., device='cuda:0')]

Training loss: 3563584.5

Prediction time: [datetime.timedelta(microseconds=78666), datetime.timedelta(microseconds=73672), datetime.timedelta(microseconds=75678), datetime.timedelta(microseconds=74682), datetime.timedelta(microseconds=77671), datetime.timedelta(microseconds=79662), datetime.timedelta(microseconds=77671), datetime.timedelta(microseconds=77655), datetime.timedelta(microseconds=78667), datetime.timedelta(microseconds=76674)]

Phi time: [datetime.timedelta(seconds=1, microseconds=419134), datetime.timedelta(microseconds=854592), datetime.timedelta(microseconds=842822), datetime.timedelta(microseconds=860720), datetime.timedelta(microseconds=852385), datetime.timedelta(microseconds=852463), datetime.timedelta(microseconds=877478), datetime.timedelta(microseconds=857887), datetime.timedelta(microseconds=854012), datetime.timedelta(microseconds=848238)]

