Shiki42's picture
insert_key single-task dev3 checkpoint (178ee15, 96.9% deterministic mixed-reset eval) and its reset caches
b292463 verified
Raw History Blame Contribute Delete
6.13 kB
{
"args": {
"mode": "train",
"config": "a2r/benchmark/contact_rl/configs/single-insert_key.yaml",
"output": "/home/coder/contact_rl/runs/insert_key-rl-dev3",
"sim_python": "/home/coder/agentic-robodojo/venv/bin/python",
"cache_root": "/home/coder/contact_rl/caches",
"minutes": 90.0,
"max_iters": 10000000,
"curriculum": false,
"promote_at": 0.9,
"lr": 0.0003,
"entropy": 0.0,
"horizon": 32,
"checkpoint": "",
"rounds": 2,
"instances": "train",
"seed": 0,
"trace_envs": 0
},
"config": {
"tasks": [
{
"task": "insert_key",
"num_envs": 1024,
"cache": "insert_key/cache.pt",
"heldout_cache": "insert_key/cache_heldout.pt"
}
]
},
"vocab": {
"capacity": {
"interactions": 32,
"objects": 128
},
"interactions": {
"insert_turn": 0,
"insert": 1,
"screw": 2,
"hang": 3,
"sweep": 4,
"place": 5,
"fold": 6
},
"objects": {
"key": 0,
"key_slot": 1,
"charger": 2,
"socket": 3,
"nut": 4,
"bolt": 5,
"mug": 6,
"mug_rack": 7,
"block": 8,
"broom": 9,
"dustpan": 10,
"egg": 11,
"egg_holder": 12,
"cloth": 13
}
},
"tasks": [
{
"task": "insert_key",
"spec": {
"name": "insert_key",
"interaction": "insert_turn",
"mover": "key",
"target": "key_slot",
"tool": null
},
"ids": [
0,
0,
1,
-1
],
"num_envs": 1024,
"n_points": 168,
"roles": [
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
3,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1
],
"max_episode_length": 120,
"control_hz": 30.0,
"cache_entries": 72288,
"git": {
"commit": "178ee15a4e7e863b3ccd30c90f7a6b8a4480312f",
"dirty": false
},
"args": {
"mode": "serve",
"task": "insert_key",
"output": "/home/coder/contact_rl/runs/insert_key-rl-dev3/sim_0_insert_key",
"num_envs": 1024,
"cache": "/home/coder/contact_rl/caches/insert_key/cache.pt",
"rounds": 4,
"seed": 0,
"eval": false,
"instances": "train",
"max_difficulty": 1.0,
"socket": "/home/coder/contact_rl/runs/insert_key-rl-dev3/sim_0_insert_key.sock",
"piper_source": "",
"steps": 0,
"trace_envs": 0,
"set": [],
"enable_cameras": false,
"xr": false,
"device": "cuda:0",
"cpu": false,
"verbose": false,
"info": false,
"rendering_mode": null,
"kit_args": "",
"anim_recording_enabled": false,
"anim_recording_start_time": 0,
"anim_recording_stop_time": 10,
"headless": true,
"hide_ui": true,
"create_new_stage": false,
"physics_gpu": 0,
"active_gpu": 0
},
"python": "3.11.15 (main, Jun 2 2026, 22:26:03) [Clang 22.1.3 ]"
}
],
"environment_check": {
"interpreter": "/home/coder/contact_rl/runtime/learner312/bin/python",
"python": "3.12.13 (main, Jun 2 2026, 22:27:15) [Clang 22.1.3 ]",
"python_major_minor": [
3,
12
],
"packages": {
"lerobot": {
"version": "0.6.1",
"import_path": "/home/coder/contact_rl/runtime/learner312/lib/python3.12/site-packages/lerobot/__init__.py",
"dist_path": "/home/coder/contact_rl/runtime/learner312/lib/python3.12/site-packages",
"direct_url": null
},
"transformers": {
"version": "5.4.0",
"import_path": "/home/coder/contact_rl/runtime/learner312/lib/python3.12/site-packages/transformers/__init__.py",
"dist_path": "/home/coder/contact_rl/runtime/learner312/lib/python3.12/site-packages",
"direct_url": null
},
"torch": {
"version": "2.11.0+cu128",
"import_path": "/home/coder/contact_rl/runtime/learner312/lib/python3.12/site-packages/torch/__init__.py",
"dist_path": "/home/coder/contact_rl/runtime/learner312/lib/python3.12/site-packages",
"direct_url": null
}
},
"dataset": "not_applicable: online RL, no LeRobotDataset",
"ok": true
},
"git": {
"commit": "178ee15a4e7e863b3ccd30c90f7a6b8a4480312f",
"dirty": false
}
}