| {"circle": {"means": {"training_time": 158.68333101272583, "eval_loss": 0.43554273544979827, "inference_time": 0.13546013832092285, "kl_divergence": 0.33243008770187155}}, "dino": {"means": {"training_time": 173.21343445777893, "eval_loss": 0.6627920077889776, "inference_time": 0.1271352767944336, "kl_divergence": 1.1349697420258653}}, "line": {"means": {"training_time": 169.03819370269775, "eval_loss": 0.8004888259541348, "inference_time": 0.10934042930603027, "kl_divergence": 0.16438389480429438}}, "moons": {"means": {"training_time": 177.6702971458435, "eval_loss": 0.6154349210012294, "inference_time": 0.11519336700439453, "kl_divergence": 0.11006145382375511}}} |