| {"circle": {"means": {"training_time": 37.41756200790405, "eval_loss": 0.43862981091984704, "inference_time": 0.17150163650512695, "kl_divergence": 0.35409057707548985}}, "dino": {"means": {"training_time": 36.680198669433594, "eval_loss": 0.6648215834442002, "inference_time": 0.17148971557617188, "kl_divergence": 0.9891262038552158}}, "line": {"means": {"training_time": 37.15258550643921, "eval_loss": 0.8037568644794357, "inference_time": 0.16029620170593262, "kl_divergence": 0.16078266200244817}}, "moons": {"means": {"training_time": 36.608174562454224, "eval_loss": 0.6160634865846171, "inference_time": 0.16797804832458496, "kl_divergence": 0.08958379744366118}}} |