{"circle": {"means": {"training_time": 48.47419357299805, "eval_loss": 0.4392722546292083, "inference_time": 0.18316245079040527, "kl_divergence": 0.35930819035619976}}, "dino": {"means": {"training_time": 41.885783672332764, "eval_loss": 0.6636652672077383, "inference_time": 0.18297195434570312, "kl_divergence": 1.060376674621348}}, "line": {"means": {"training_time": 38.887343406677246, "eval_loss": 0.8017848281909132, "inference_time": 0.17120051383972168, "kl_divergence": 0.15692256311119815}}, "moons": {"means": {"training_time": 38.7231330871582, "eval_loss": 0.6203141152248968, "inference_time": 0.1772310733795166, "kl_divergence": 0.09455949519397541}}}