wassname/iris_bigvae
1
0
Fork 0
mirror of https://github.com/wassname/iris_bigvae.git synced 2026-09-09 11:24:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

document play.sh options

Browse Source
This commit is contained in:
wassname
2023-11-18 16:32:59 +08:00
parent a341e7aaad
commit 9b399031ca
3 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/play.sh
+7 -7
View File
@@ -14,22 +14,22 @@ while [ "$1" != "" ]; do
;;
-h | --header )
header=1
;;
;; # adds banner with env metadata like action
-r | --reconstruction )
reconstruction=1
;;
;; # 3 panes [original_obs, resized_obs, reconstructed], doesn't do anything if any of -w -a or -e are set. shows quality of encoder decoder
-s | --save-mode )
save_mode=1
;;
;; # lets you save the episode to mp4
-a | --agent-world-model )
mode="agent_in_world_model"
;;
;; # the agent plays in the world model env, shows the quality of the dynamics model
-e | --episode )
mode="episode_replay"
;;
;; # replay train, test, or imagined episodes. shows quality of dynamics model
-w | --world-model )
mode="play_in_world_model"
;;
;; # human plays in world model
* )
echo Invalid usage : $1
exit 1
@@ -37,4 +37,4 @@ while [ "$1" != "" ]; do
shift
done
python src/play.py hydra.run.dir=. hydra.output_subdir=null +mode="${mode}" +fps="${fps}" +header="${header}" +reconstruction="${reconstruction}" +save_mode="${save_mode}"
python -m pdb src/play.py hydra.run.dir=. hydra.output_subdir=null +mode="${mode}" +fps="${fps}" +header="${header}" +reconstruction="${reconstruction}" +save_mode="${save_mode}"