{"circle": {"means": {"training_time": 320.02629947662354, "eval_loss": 0.38405951567927893, "inference_time": 0.13506627082824707, "kl_divergence": 0.40717430955596984}}, "dino": {"means": {"training_time": 321.308536529541, "eval_loss": 0.6355158123366363, "inference_time": 0.1341416835784912, "kl_divergence": 0.536219988725031}}, "line": {"means": {"training_time": 322.25109910964966, "eval_loss": 0.8613292065727741, "inference_time": 0.13298535346984863, "kl_divergence": 0.33859042032664705}}, "moons": {"means": {"training_time": 301.97175002098083, "eval_loss": 0.6417264534384394, "inference_time": 0.13048815727233887, "kl_divergence": 0.2174388676156737}}}