maze.sh

#!/bin/bash

#SBATCH --mem=50G
#SBATCH -c 8
#SBATCH --gres=gpu:1
#SBATCH -p gpu


CKPT=/checkpoint/minghan/${SLURM_JOB_ID}
touch $CKPT
python dreamer.py \
		--load_config True \
		--logdir $1 \
		--ckptdir $CKPT \
		--task $2 \
		--expl_behavior $3 \
		--use_wm $4 \
		--action_repeat $5 > $1/train.log 2>& 1