diff --git a/scripts/slurm_run.sh b/scripts/slurm_run.sh index cebf651..a8080bf 100755 --- a/scripts/slurm_run.sh +++ b/scripts/slurm_run.sh @@ -3,22 +3,21 @@ # Example ussage: # # sbatch ./scripts/slurm_run.sh scripts/train_baseline.sh \ -# --train.run_name=realikun_16x8_0001 +# --run-name=test --wandb-project=nmmo --wandb-entity=kywch #SBATCH --account=carperai -#SBATCH --partition=g40 +#SBATCH --partition=g40x #SBATCH --nodes=1 #SBATCH --gpus=1 -#SBATCH --cpus-per-gpu=12 +#SBATCH --cpus-per-gpu=6 #__SBATCH --mem=80G -#SBATCH --chdir=/fsx/home-daveey/nmmo-baselines/ +#SBATCH --chdir=/fsx/proj-nmmo/nmmo-baselines/ #SBATCH --output=sbatch/%j.log #SBATCH --error=sbatch/%j.log #SBATCH --requeue -#SBATCH --export=PYTHONUNBUFFERED=1,WANDB_BASE_URL="https://stability.wandb.io",WANDB_DIR=/fsx/home-daveey/tmp/wandb,WANDB_CONFIG_DIR=/fsx/home-daveey/tmp/wandb +#SBATCH --export=PYTHONUNBUFFERED=1,WANDB_BASE_URL="https://stability.wandb.io",WANDB_DIR=/fsx/proj-nmmo/tmp/wandb,WANDB_CONFIG_DIR=/fsx/proj-nmmo/tmp/wandb -source /fsx/home-daveey/miniconda/etc/profile.d/conda.sh && \ -conda activate nmmo && \ +source /fsx/proj-nmmo/venv/bin/activate && \ ulimit -c unlimited && \ ulimit -s unlimited && \ ulimit -a diff --git a/scripts/train_baseline.sh b/scripts/train_baseline.sh index 32464ac..a7d2a56 100755 --- a/scripts/train_baseline.sh +++ b/scripts/train_baseline.sh @@ -1,14 +1,5 @@ #!/bin/bash -python -u -O -m train \ ---wandb.entity=daveey \ ---wandb.project=nmmo \ ---train.runs_dir=/fsx/home-daveey/runs/ \ ---train.num_steps=10000000000 \ ---env.num_maps=100 \ ---rollout.num_buffers=2 \ ---rollout.num_envs=12 \ ---env.num_npcs=266 \ ---ppo.training_batch_size=128 \ ---rollout.batch_size=131072 \ +python -u -m train \ +--runs-dir=/fsx/proj-nmmo/runs/ \ "${@}"