Skip to content

Instantly share code, notes, and snippets.

@alexlimh

alexlimh/maze.sh

Created May 17, 2020 04:42
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save alexlimh/19c820b889033b0f91d10c6ee42d3216 to your computer and use it in GitHub Desktop.
Save alexlimh/19c820b889033b0f91d10c6ee42d3216 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment