forked from Lifelong-Robot-Learning/LIBERO
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathexecute_fine.sh
executable file
·23 lines (18 loc) · 1.27 KB
/
execute_fine.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
sbatch slurm/train.sbatch python libero/lifelong/few_shot.py \
pretrain_model_path="/storage/home/hcoda1/0/amete7/p-agarg35-0/diff-skill/LIBERO/experiments_clip/LIBERO_90/Multitask/SkillGPT_Model/ResnetEncoder/m4op_32_f4_k3s4_tt_n6d384_off0/run_001/multitask_model_ep100.pth" \
exp_name="m4op_32_f4_k3s4_tt_n6d384_off0_5shot" \
sbatch slurm/train.sbatch python libero/lifelong/few_shot.py \
pretrain_model_path="/storage/home/hcoda1/0/amete7/p-agarg35-0/diff-skill/LIBERO/experiments_clip/LIBERO_90/Multitask/SkillGPT_Model/ResnetEncoder/m4op_8_f4_k3s4_tt_n6d384_off100/run_001/multitask_model_ep100.pth" \
exp_name="m4op_8_f4_k3s4_tt_n6d384_off100_5shot" \
data.seq_len=8 \
policy.skill_block_size=8 \
policy.prior.block_size=2 \
policy.offset_loss_scale=100 \
sbatch slurm/train.sbatch python libero/lifelong/few_shot.py \
pretrain_model_path="/storage/home/hcoda1/0/amete7/p-agarg35-0/diff-skill/LIBERO/experiments_clip/LIBERO_90/Multitask/SkillGPT_Model/ResnetEncoder/m4op_32_f5_k3s4_tt_n6d384/run_001/multitask_model_ep100.pth" \
exp_name="m4op_32_f5_k3s4_tt_n6d384_5shot" \
policy.fsq_level=[7,5,5,5,5] \
policy.offset_loss_scale=1 \
policy.prior.vocab_size=4380 \
policy.prior.output_dim=4375 \
policy.prior.start_token=4376 \