| {"circle": {"means": {"training_time": 73.97696614265442, "eval_loss": 0.43658909529371337, "inference_time": 0.2919645309448242, "kl_divergence": 0.3918276454964591}}, "dino": {"means": {"training_time": 72.91716957092285, "eval_loss": 0.6578772964379976, "inference_time": 0.2855503559112549, "kl_divergence": 0.583503367599638}}, "line": {"means": {"training_time": 73.47489547729492, "eval_loss": 0.8102090942585255, "inference_time": 0.285874605178833, "kl_divergence": 0.19102262295215622}}, "moons": {"means": {"training_time": 77.20445489883423, "eval_loss": 0.6149405840870059, "inference_time": 0.2781538963317871, "kl_divergence": 0.07946220210924662}}} |