tweak slurm scripts

CarperAI · Mar 29, 2024 · 942da4d · 942da4d
1 parent 02fb486
commit 942da4d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/config.yaml b/config.yaml
@@ -1,5 +1,5 @@
 wandb:
-  project: nmmo
+  project: nmmo-baselines
   entity: kywch
   group: ~
 

diff --git a/scripts/slurm_run.sh b/scripts/slurm_run.sh
@@ -9,20 +9,20 @@
 #SBATCH --partition=g40x
 #SBATCH --nodes=1
 #SBATCH --gpus=1
-#SBATCH --cpus-per-gpu=6
+#SBATCH --cpus-per-gpu=8
 #__SBATCH --mem=80G
-#SBATCH --chdir=/fsx/proj-nmmo/nmmo-baselines/
+#SBATCH --chdir=/weka/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/proj-nmmo/tmp/wandb,WANDB_CONFIG_DIR=/fsx/proj-nmmo/tmp/wandb
+#SBATCH --export=PYTHONUNBUFFERED=1,WANDB_DIR=/weka/proj-nmmo/tmp/wandb,WANDB_CONFIG_DIR=/weka/proj-nmmo/tmp/wandb
 
-source /fsx/proj-nmmo/venv/bin/activate && \
+source /weka/proj-nmmo/venv/bin/activate && \
 ulimit -c unlimited && \
 ulimit -s unlimited && \
 ulimit -a
 
-wandb login --host=https://stability.wandb.io
+wandb login --cloud
 
 # Extract run_name from the arguments
 run_name=""

diff --git a/scripts/train_baseline.sh b/scripts/train_baseline.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
 
 python -u -m train \
---runs-dir=/fsx/proj-nmmo/runs/ \
+--train.data_dir=/weka/proj-nmmo/runs/ \
 "${@}"