File size: 5,586 Bytes
aafb370 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | {
"accumulate_grad_batches": 1,
"backbone": "clip",
"base_log_dir": {
"_string": "train_logs",
"_type": "python_object (type = PosixPath)",
"_value": "gASVKgAAAAAAAACMB3BhdGhsaWKUjAlQb3NpeFBhdGiUk5SMCnRyYWluX2xvZ3OUhZRSlC4="
},
"batch_size": 8,
"batch_size_val": 1,
"cache_size": 0,
"cache_size_val": 0,
"cameras": {
"_type": "tuple",
"_value": [
"front",
"wrist"
]
},
"checkpoint": {
"_string": "/work/popo9790/train_log/rep_ctx/rep_ctx-192-B8-accu2-ngpus2-lr3e-4-DI1-20-H3-DT25-backboneclip-S256,256-R1-wd5e-3/last.pth",
"_type": "python_object (type = PosixPath)",
"_value": "gASVqQAAAAAAAACMB3BhdGhsaWKUjAlQb3NpeFBhdGiUk5QojAEvlIwEd29ya5SMCHBvcG85NzkwlIwJdHJhaW5fbG9nlIwHcmVwX2N0eJSMUXJlcF9jdHgtMTkyLUI4LWFjY3UyLW5ncHVzMi1scjNlLTQtREkxLTIwLUgzLURUMjUtYmFja2JvbmVjbGlwLVMyNTYsMjU2LVIxLXdkNWUtM5SMCGxhc3QucHRolHSUUpQu"
},
"dataset": {
"_string": "/work/popo9790/dataset/packaged_ABC_D_ours/training",
"_type": "python_object (type = PosixPath)",
"_value": "gASVXwAAAAAAAACMB3BhdGhsaWKUjAlQb3NpeFBhdGiUk5QojAEvlIwEd29ya5SMCHBvcG85NzkwlIwHZGF0YXNldJSME3BhY2thZ2VkX0FCQ19EX291cnOUjAh0cmFpbmluZ5R0lFKULg=="
},
"dense_interpolation": 1,
"diffusion_timesteps": 25,
"embedding_dim": 192,
"end_lr": 1e-06,
"eval_only": 0,
"exp_log_dir": "Planner_Calvin",
"fps_subsampling_factor": 3,
"gripper_loc_bounds": {
"_string": "[[-0.30205889 -0.25688798 -0.21557967]\n [ 0.36127854 0.33770842 0.18373556]]",
"_type": "python_object (type = ndarray)",
"_value": "gASVugAAAAAAAACMFW51bXB5LmNvcmUubXVsdGlhcnJheZSMDF9yZWNvbnN0cnVjdJSTlIwFbnVtcHmUjAduZGFycmF5lJOUSwCFlEMBYpSHlFKUKEsBSwJLA4aUaAOMBWR0eXBllJOUjAJmOJSJiIeUUpQoSwOMATyUTk5OSv////9K/////0sAdJRiiUMwpHA9yu5U07+kcD1K2nDQv0jhelQdmMu/pHA9CjAf1z+kcD3KA53VP0jhepSlhMc/lHSUYi4="
},
"gripper_loc_bounds_buffer": 0.01,
"image_rescale": "0.75,1.25",
"image_size": "256,256",
"instructions": {
"_string": "instructions/calvin_task_ABC_D",
"_type": "python_object (type = PosixPath)",
"_value": "gASVQAAAAAAAAACMB3BhdGhsaWKUjAlQb3NpeFBhdGiUk5SMDGluc3RydWN0aW9uc5SMEWNhbHZpbl90YXNrX0FCQ19ElIaUUpQu"
},
"interpolation_length": 20,
"keypose_only": 0,
"lang_enhanced": 1,
"local_rank": 0,
"log_dir": {
"_string": "/work/popo9790/train_log/rep_ctx/rep_ctx-192-B8-accu2-ngpus2-lr3e-4-DI1-20-H3-DT25-backboneclip-S256,256-R1-wd5e-3",
"_type": "python_object (type = PosixPath)",
"_value": "gASVngAAAAAAAACMB3BhdGhsaWKUjAlQb3NpeFBhdGiUk5QojAEvlIwEd29ya5SMCHBvcG85NzkwlIwJdHJhaW5fbG9nlIwHcmVwX2N0eJSMUXJlcF9jdHgtMTkyLUI4LWFjY3UyLW5ncHVzMi1scjNlLTQtREkxLTIwLUgzLURUMjUtYmFja2JvbmVjbGlwLVMyNTYsMjU2LVIxLXdkNWUtM5R0lFKULg=="
},
"lr": 0.0003,
"lr_decay": 0.9999,
"max_episode_length": 30,
"max_episodes_per_task": -1,
"noise_end": 1.0,
"noise_power": 3.0,
"noise_start": 1e-08,
"num_history": 3,
"num_joints": 7,
"num_vis_ins_attn_layers": 2,
"num_workers": 4,
"quaternion_format": "wxyz",
"relative_action": 1,
"reproducibility": {
"command_line": "python main_trajectory_calvin.py --tasks A B C D --backbone clip --dataset /work/popo9790/dataset/packaged_ABC_D_ours/training --valset /work/popo9790/dataset/packaged_ABC_D_ours/validation --instructions instructions/calvin_task_ABC_D/ --gripper_loc_bounds tasks/calvin_rel_traj_location_bounds_task_ABC_D.json --gripper_loc_bounds_buffer 0.01 --image_size 256,256 --num_workers 4 --max_episode_length 30 --train_iters 44000 --embedding_dim 192 --use_instruction 1 --rotation_parametrization 6D --diffusion_timesteps 25 --val_freq 1000 --val_iters 2 --dense_interpolation 1 --interpolation_length 20 --exp_log_dir Planner_Calvin --batch_size 8 --batch_size_val 1 --cache_size 0 --cache_size_val 0 --keypose_only 0 --lr 3e-4 --end_lr 1e-6 --lr_decay 0.9999 --wd 5e-3 --num_history 3 --cameras front wrist --max_episodes_per_task -1 --relative_action 1 --fps_subsampling_factor 3 --lang_enhanced 1 --quaternion_format wxyz --run_log_dir /work/popo9790/train_log/rep_ctx/rep_ctx-192-B8-accu2-ngpus2-lr3e-4-DI1-20-H3-DT25-backboneclip-S256,256-R1-wd5e-3 --checkpoint /work/popo9790/train_log/rep_ctx/rep_ctx-192-B8-accu2-ngpus2-lr3e-4-DI1-20-H3-DT25-backboneclip-S256,256-R1-wd5e-3/last.pth",
"git_has_uncommitted_changes": true,
"git_root": "/home/popo9790/rep_ctx",
"git_url": "https://github.com/tars3017/CoG/tree/45807b1f6b6382e8e10209c2a7551ba206805439",
"time": "Sat 29 Nov 2025 09:39:33 PM CST"
},
"rotation_parametrization": "6D",
"run_log_dir": "/work/popo9790/train_log/rep_ctx/rep_ctx-192-B8-accu2-ngpus2-lr3e-4-DI1-20-H3-DT25-backboneclip-S256,256-R1-wd5e-3",
"seed": 0,
"tasks": {
"_type": "tuple",
"_value": [
"A",
"B",
"C",
"D"
]
},
"train_iters": 44000,
"use_instruction": 1,
"val_freq": 1000,
"val_iters": 2,
"valset": {
"_string": "/work/popo9790/dataset/packaged_ABC_D_ours/validation",
"_type": "python_object (type = PosixPath)",
"_value": "gASVYQAAAAAAAACMB3BhdGhsaWKUjAlQb3NpeFBhdGiUk5QojAEvlIwEd29ya5SMCHBvcG85NzkwlIwHZGF0YXNldJSME3BhY2thZ2VkX0FCQ19EX291cnOUjAp2YWxpZGF0aW9ulHSUUpQu"
},
"variations": {
"_type": "tuple",
"_value": [
0
]
},
"wd": 0.005,
"weight_decay": 1e-12
} |