{ "dataset": "peg-insert-hrate", "sensor": "vbts", "model_type": "ImageAutoencoder", "latent_dim": 16, "input_shape": [ 3, 64, 64 ], "external_val_loss": 0.0012740151165832045, "subtract_first_frame": false, "train_episodes": [ 0, 4, 5, 8, 9, 10, 12, 21, 25, 28, 32, 33, 35 ], "val_episodes": [ 41, 42, 43, 44, 45, 46, 47, 48, 49, 50 ] }