#!/bin/bash

set -e
set -o pipefail
cd /path/to/run/and/test/shell/scripts
game="MontezumaRevengeNoFrameskip-v4"
outputDir="/my/experiment/output/dir"
timesteps=640
alg="a2c"
./baselines_train.sh $alg $game $timesteps $outputDir --cs true --csGridSize 16 --csWeight 0.75 --retrain true --checkpointInterval 4000000
mv $outputDir/log.txt $outputDir/training.txt
./baselines_test.sh $alg $game $timesteps $outputDir 50000 --cs true --csGridSize 16 --csWeight 0.75 --createVideo true
