mirror of
https://github.com/wassname/ray.git
synced 2026-08-18 12:20:14 +08:00
[rllib] Added evaluation script to RLLib (#1295)
This commit is contained in:
committed by
Richard Liaw
parent
96c46d35ff
commit
20d6b74aa6
@@ -0,0 +1,23 @@
|
||||
#!/bin/sh
|
||||
|
||||
# TODO: Test AC3
|
||||
ALGS='DQN PPO'
|
||||
GYM_ENV='CartPole-v0'
|
||||
|
||||
for ALG in $ALGS
|
||||
do
|
||||
EXPERIMENT_NAME=$GYM_ENV'_'$ALG
|
||||
python /ray/python/ray/rllib/train.py --run $ALG --env $GYM_ENV \
|
||||
--stop '{"training_iteration": 2}' --experiment-name $EXPERIMENT_NAME \
|
||||
--checkpoint-freq 1
|
||||
|
||||
EXPERIMENT_PATH='/tmp/ray/'$EXPERIMENT_NAME
|
||||
CHECKPOINT_FOLDER=$(ls $EXPERIMENT_PATH)
|
||||
CHECKPOINT=$EXPERIMENT_PATH'/'$CHECKPOINT_FOLDER'/checkpoint-1'
|
||||
|
||||
python /ray/python/ray/rllib/eval.py $CHECKPOINT --run $ALG \
|
||||
--env $GYM_ENV --no-render
|
||||
|
||||
# Clean up
|
||||
rm -rf $EXPERIMENT_PATH
|
||||
done
|
||||
Reference in New Issue
Block a user