Skip to content

Commit

Permalink
Train_PPO script
Browse files Browse the repository at this point in the history
  • Loading branch information
Gregwar committed Mar 18, 2024
1 parent d5101dd commit 5229fa0
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions train_ppo.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
#!/bin/bash

if [ -z "$1" ]; then
echo "Usage: $0 <run_name>"
exit 1
fi

PROJECT=sigmaban-standup
ENV=sigmaban-standup-v0
ALGO=ppo
TAGS=standup
EXP_ID=`python next_run_id.py $ALGO $ENV`
RUN_NAME="${HOSTNAME}_${EXP_ID}_$1"

# killall -9 python

nohup python train.py --run-name="$RUN_NAME" \
--env=$ENV --algo=$ALGO --track --wandb-project-name $PROJECT \
--eval-freq=150000 \
--eval-episodes=100 --n-eval-envs=4 --vec-env=subproc \
-tags exp-$EXP_ID $TAGS h-$HOSTNAME > $HOSTNAME.out &

tail -f $HOSTNAME.out

0 comments on commit 5229fa0

Please sign in to comment.