run_lider_bc.sh

#!/bin/bash

# Six games to choose from
game=$1

if [ "$game" == "Gopher" ]; then
    echo "Training Gopher..."
elif [ "$game" == "NameThisGame" ]; then
    echo "Training NameThisGame..."
elif [ "$game" == "MsPacman" ]; then
    echo "Training Ms.Pac-Man..."
elif [ "$game" == "Alien" ]; then
    echo "Training Alien..."
elif [ "$game" == "Freeway" ]; then
    echo "Training Freeway..."
elif [ "$game" == "MontezumaRevenge" ]; then
    echo "Training Montezuma's Revenge..."
else
    echo "invalid game!"
    echo "Choose from [Gopher, NameThisGame, MsPacman, Alien, Freeway, MontezumaRevenge]."
    echo "For example: ./run_lider.sh MsPacman"
    echo "Exiting..."; exit
fi

python3 LiDER/run_experiment.py \
  --gym-env=${game}NoFrameskip-v4 \
  --parallel-size=17 \
  --max-time-step-fraction=0.5 \
  --use-mnih-2015 --input-shape=88 --padding=SAME \
  --unclipped-reward --transformed-bellman \
  --use-sil --priority-memory \
  --batch-size=32 \
  --checkpoint-buffer --checkpoint-freq=1 \
  --append-experiment-num=1 \
  --use-lider \
  --load-BC \
  --pretrained-model-folder=pretrained_models/BC/${game}/${game}NoFrameskip_v4_checkpoint \
  # --use-gpu --cuda-devices=0 \ # uncomment for gpu support