{"circle": {"means": {"training_time": 72.7797212600708, "eval_loss": 0.44442242086695893, "inference_time": 0.2980952262878418, "kl_divergence": 0.47009555896972094}}, "dino": {"means": {"training_time": 75.91083240509033, "eval_loss": 0.6673849075651535, "inference_time": 0.29502367973327637, "kl_divergence": 0.6495770647785007}}, "line": {"means": {"training_time": 77.7726686000824, "eval_loss": 0.8133890747719104, "inference_time": 0.28652405738830566, "kl_divergence": 0.2489773415001416}}, "moons": {"means": {"training_time": 70.94407176971436, "eval_loss": 0.6255804364333677, "inference_time": 0.2740786075592041, "kl_divergence": 0.11055475645165658}}}