| {"circle": {"means": {"training_time": 73.06966805458069, "eval_loss": 0.43969630813964494, "inference_time": 0.29320263862609863, "kl_divergence": 0.3689575513483317}}, "dino": {"means": {"training_time": 74.27817940711975, "eval_loss": 0.6613499774499927, "inference_time": 0.2861502170562744, "kl_divergence": 0.8196823128731071}}, "line": {"means": {"training_time": 76.55267119407654, "eval_loss": 0.8027192704817828, "inference_time": 0.274810791015625, "kl_divergence": 0.1723356430884586}}, "moons": {"means": {"training_time": 74.5637640953064, "eval_loss": 0.6173960363773434, "inference_time": 0.27197885513305664, "kl_divergence": 0.09956056764691522}}} |