{"circle": {"means": {"training_time": 29.467289686203003, "eval_loss": 0.4338561762171938, "inference_time": 0.14316987991333008, "kl_divergence": 0.3639135723344224}}, "dino": {"means": {"training_time": 28.34086012840271, "eval_loss": 0.6643665589944786, "inference_time": 0.07733488082885742, "kl_divergence": 1.044781905549579}}, "line": {"means": {"training_time": 26.879716634750366, "eval_loss": 0.8022636874862339, "inference_time": 0.07804012298583984, "kl_divergence": 0.15131457765124726}}, "moons": {"means": {"training_time": 25.760434865951538, "eval_loss": 0.6167483846549793, "inference_time": 0.08142733573913574, "kl_divergence": 0.0831315587640603}}}