| { | |
| "dataset": "insertpeg", | |
| "epoch": 10, | |
| "format_version": 1, | |
| "frame_skip": 5, | |
| "model": "prejepa", | |
| "sha256": "5f44b781ba9ae4a7adddf457ca84e4e7fd2c1e4d3df2d0e4bb5d8c3ffd5dcc4d", | |
| "size_bytes": 181289001, | |
| "source_config_filename": "config.yaml", | |
| "source_filename": "weights_epoch_10.pt", | |
| "training_config": { | |
| "batch_size": 32, | |
| "frameskip": 5, | |
| "image_size": 224, | |
| "model.predictor.num_frames": "${wm.history_size}", | |
| "optimizer.lr": 0.0005, | |
| "optimizer.weight_decay": 0.0, | |
| "seed": 42, | |
| "trainer.max_epochs": 10 | |
| }, | |
| "training_run": "20260911_manipulator_insert_peg_prejepa_rgb_action_fs5_3173369" | |
| } | |