remove unecessary files
This commit is contained in:
parent
f1cd0d83d2
commit
3122ecbbfb
|
@ -1,79 +0,0 @@
|
||||||
{
|
|
||||||
"type": "pi0fast",
|
|
||||||
"n_obs_steps": 1,
|
|
||||||
"normalization_mapping": {
|
|
||||||
"VISUAL": "IDENTITY",
|
|
||||||
"STATE": "MEAN_STD",
|
|
||||||
"ACTION": "MEAN_STD"
|
|
||||||
},
|
|
||||||
"input_features": {
|
|
||||||
"observation.image": {
|
|
||||||
"type": "VISUAL",
|
|
||||||
"shape": [
|
|
||||||
3,
|
|
||||||
256,
|
|
||||||
256
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"observation.image2": {
|
|
||||||
"type": "VISUAL",
|
|
||||||
"shape": [
|
|
||||||
3,
|
|
||||||
256,
|
|
||||||
256
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"observation.image3": {
|
|
||||||
"type": "VISUAL",
|
|
||||||
"shape": [
|
|
||||||
3,
|
|
||||||
256,
|
|
||||||
256
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"observation.state": {
|
|
||||||
"type": "STATE",
|
|
||||||
"shape": [
|
|
||||||
8
|
|
||||||
]
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"output_features": {
|
|
||||||
"action": {
|
|
||||||
"type": "ACTION",
|
|
||||||
"shape": [
|
|
||||||
7
|
|
||||||
]
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"chunk_size": 10,
|
|
||||||
"n_action_steps": 5,
|
|
||||||
"max_state_dim": 32,
|
|
||||||
"max_action_dim": 32,
|
|
||||||
"resize_imgs_with_padding": [
|
|
||||||
224,
|
|
||||||
224
|
|
||||||
],
|
|
||||||
"interpolate_like_pi": false,
|
|
||||||
"empty_cameras": 0,
|
|
||||||
"adapt_to_pi_aloha": false,
|
|
||||||
"use_delta_joint_actions_aloha": false,
|
|
||||||
"tokenizer_max_length": 48,
|
|
||||||
"proj_width": 1024,
|
|
||||||
"max_decoding_steps": 256,
|
|
||||||
"fast_skip_tokens": 128,
|
|
||||||
"max_input_seq_len": 256,
|
|
||||||
"use_cache": true,
|
|
||||||
"freeze_vision_encoder": true,
|
|
||||||
"freeze_lm_head": true,
|
|
||||||
"optimizer_lr": 0.0001,
|
|
||||||
"optimizer_betas": [
|
|
||||||
0.9,
|
|
||||||
0.95
|
|
||||||
],
|
|
||||||
"optimizer_eps": 1e-08,
|
|
||||||
"optimizer_weight_decay": 1e-5,
|
|
||||||
"scheduler_warmup_steps": 1000,
|
|
||||||
"scheduler_decay_steps": 30000,
|
|
||||||
"scheduler_decay_lr": 2.5e-06
|
|
||||||
}
|
|
|
@ -1,27 +0,0 @@
|
||||||
cd ~/lerobot
|
|
||||||
source ~/miniconda3/bin/activate
|
|
||||||
conda activate lerobot
|
|
||||||
|
|
||||||
export MUJOCO_GL=egl
|
|
||||||
|
|
||||||
ENV=aloha
|
|
||||||
TASK=AlohaTransferCube-v0
|
|
||||||
REPO_ID=lerobot/aloha_sim_transfer_cube_human
|
|
||||||
OUT_DIR=~/logs/lerobot/tmp/act_aloha_transfer
|
|
||||||
|
|
||||||
EVAL_FREQ=5000
|
|
||||||
OFFLINE_STEP=30000
|
|
||||||
SAVE_FREQ=100000
|
|
||||||
|
|
||||||
POLICY_PATH=~/.cache/openpi/openpi-assets/checkpoints/pi0_fast_base_pytorch/
|
|
||||||
POLICY=pi0fast
|
|
||||||
|
|
||||||
python lerobot/scripts/train.py \
|
|
||||||
--policy.path=$POLICY_PATH \
|
|
||||||
--dataset.repo_id=$REPO_ID \
|
|
||||||
--env.type=$ENV \
|
|
||||||
--env.task=$TASK \
|
|
||||||
--output_dir=$OUT_DIR \
|
|
||||||
--eval_freq=$EVAL_FREQ \
|
|
||||||
--steps=$OFFLINE_STEP \
|
|
||||||
--save_freq=$SAVE_FREQ
|
|
Loading…
Reference in New Issue