{"circle": {"means": {"training_time": 265.2883794307709, "eval_loss": 0.434587472006488, "inference_time": 0.14093685150146484, "kl_divergence": 0.3604047182045786}}, "dino": {"means": {"training_time": 243.74999475479126, "eval_loss": 0.665170654463951, "inference_time": 0.11144709587097168, "kl_divergence": 1.0364102334466179}}, "line": {"means": {"training_time": 261.86758947372437, "eval_loss": 0.8037817526961226, "inference_time": 0.1273975372314453, "kl_divergence": 0.14531509758172917}}, "moons": {"means": {"training_time": 263.7597990036011, "eval_loss": 0.6178049460396438, "inference_time": 0.1425645351409912, "kl_divergence": 0.10209078394617543}}}