|
|
@ -0,0 +1,54 @@ |
|
|
|
#!/bin/bash |
|
|
|
|
|
|
|
. ../env/bin/activate |
|
|
|
|
|
|
|
commit_message="$(git log -1 --oneline | sed 's/\s/_/g')" |
|
|
|
git diff --quiet || commit_message="${commit_message}_dirty" |
|
|
|
datetime="$(date +%Y%m%d_%H%M%S)" |
|
|
|
|
|
|
|
exp_name="${commit_message}-${datetime}" |
|
|
|
experiment_log_dir="${2}/${exp_name}" |
|
|
|
|
|
|
|
#cpu=nehalem192g0 |
|
|
|
cpu=opteron192g0 |
|
|
|
#cpu=haswell378g0 |
|
|
|
gpu=cuda |
|
|
|
|
|
|
|
env=$4 |
|
|
|
shielding=$5 |
|
|
|
num_steps=$1 |
|
|
|
experiment_log_dir="${2}" |
|
|
|
num_evaluations=$3 |
|
|
|
shield_value=$6 |
|
|
|
prism_config=$7 |
|
|
|
prob_next=$8 |
|
|
|
prob_direct=$9 |
|
|
|
prob_forward="${10}" |
|
|
|
shield_comparision="${11}" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# echo $experiment_log_dir |
|
|
|
# echo $(pwd) |
|
|
|
# echo $(pwd)/$experiment_log_dir/$exp_name |
|
|
|
# python3 examples/shields/rl/11_minigridrl.py --expname "$exp_name" --steps "$1" --log_dir "$experiment_log_dir"/ --evaluations "$3" --env "$4" --shielding "$5" --shield_comparision "$6" --prism_config "$7" --prob_next "$8" --prob_direct "$9" --prob_forward "${10}" --shield_value "${11}" & |
|
|
|
|
|
|
|
set -x |
|
|
|
srun -w $cpu python3 examples/shields/rl/15_train_eval_tune.py --env $4 --shielding $5 --steps $1 --expname "$exp_name" --log_dir "$experiment_log_dir" --evaluations $3 --grid_to_prism_binary_path /workstore/tknoll/Minigrid2PRISM/build/main --shield_comparision $6 --prism_config $7 --prob_next $8 --prob_direct $9 --prob_forward "${10}" --shield_comparision $6 --shield_value "${11}" & |
|
|
|
set +x |
|
|
|
|
|
|
|
sleep 20 |
|
|
|
rsync -avtr --stats $(pwd)/$experiment_log_dir/$exp_name tensorboard:/media/data1/easy_rl_tb_logs |
|
|
|
sleep 60 |
|
|
|
while [[ -n $(jobs -r) ]]; do |
|
|
|
rsync -avtr --append --stats $(pwd)/$experiment_log_dir/$exp_name tensorboard:/media/data1/easy_rl_tb_logs |
|
|
|
sleep 60; |
|
|
|
done |
|
|
|
rsync -avtr --append --stats $(pwd)/$experiment_log_dir/$exp_name tensorboard:/media/data1/easy_rl_tb_logs |
|
|
|
|
|
|
|
|
|
|
|
#!/bin/bash |
|
|
|
|
|
|
|
. ../env/bin/activate |
|
|
|
|
|
|
|
|