| {"circle": {"means": {"training_time": 273.7930860519409, "eval_loss": 0.4346648931808179, "inference_time": 0.12023258209228516, "kl_divergence": 0.3500326621068686}}, "dino": {"means": {"training_time": 253.12539935112, "eval_loss": 0.6642400203153606, "inference_time": 0.12917494773864746, "kl_divergence": 1.0426551796727075}}, "line": {"means": {"training_time": 281.75747752189636, "eval_loss": 0.8052869840046329, "inference_time": 0.12714266777038574, "kl_divergence": 0.18203549509286743}}, "moons": {"means": {"training_time": 283.6050419807434, "eval_loss": 0.6191892492039429, "inference_time": 0.1302032470703125, "kl_divergence": 0.09779758687759507}}} |