diff --git a/data/envs/metaworld/generate_dataset_all.sh b/data/envs/metaworld/generate_dataset_all.sh | |
index 802bf5c..7fe0781 100755 | |
--- a/data/envs/metaworld/generate_dataset_all.sh | |
+++ b/data/envs/metaworld/generate_dataset_all.sh | |
#!/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 | |
diff --git a/data/envs/metaworld/push_all.sh b/data/envs/metaworld/push_all.sh | |
index 9d71467..26f8933 100755 | |
--- a/data/envs/metaworld/push_all.sh | |
+++ b/data/envs/metaworld/push_all.sh | |
#!/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 | |
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 | |
#!/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 | |
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 | |