diff --git a/data/envs/metaworld/train_all.sh b/data/envs/metaworld/train_all.sh index dbf328a..071e16a 100755 --- a/data/envs/metaworld/train_all.sh +++ b/data/envs/metaworld/train_all.sh @@ -1,29 +1,29 @@ #!/bin/bash ENVS=( - assembly - basketball - bin-picking - box-close - button-press-topdown - button-press-topdown-wall - button-press - button-press-wall - coffee-button - coffee-pull - coffee-push - dial-turn - disassemble - door-close - door-lock - door-open - door-unlock - drawer-close - drawer-open - faucet-close - faucet-open - hammer - hand-insert + # assembly + # basketball + # bin-picking + # box-close + # button-press-topdown + # button-press-topdown-wall + # button-press + # button-press-wall + # coffee-button + # coffee-pull + # coffee-push + # dial-turn + # disassemble + # door-close + # door-lock + # door-open + # door-unlock + # drawer-close + # drawer-open + # faucet-close + # faucet-open + # hammer + # hand-insert handle-press-side handle-press handle-pull-side @@ -55,4 +55,5 @@ ENVS=( for ENV in "${ENVS[@]}"; do python train.py --env $ENV-v2 --experiment $ENV-v2 --with_wandb True --wandb_user qgallouedec --wandb_project sample_facotry_metaworld + python enjoy.py --algo=APPO --env $ENV-v2 --experiment $ENV-v2 --train_dir=./train_dir --max_num_episodes=10 --push_to_hub --hf_repository=qgallouedec/sample-factory-$ENV-v2 --save_video --no_render --enjoy_script=enjoy --train_script=train --load_checkpoint_kind best done