sbatch script

gigantocypris · Jul 18, 2023 · 76f18ca · 76f18ca
1 parent 6706f0d
commit 76f18ca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/slurm_train.sh b/scripts/slurm_train.sh
@@ -6,7 +6,7 @@
 #SBATCH -A m2859_g       # allocation
 #SBATCH -C gpu
 #SBATCH -q regular
-#SBATCH -t 07:00:00
+#SBATCH -t 00:50:00
 #SBATCH --gpus-per-node=4
 #SBATCH --ntasks-per-gpu=1
 #SBATCH --gpus 4
@@ -18,8 +18,8 @@ export SCRATCH_FOLDER=$SCRATCH/output_PINN/$SLURM_JOB_ID
 mkdir -p $SCRATCH_FOLDER; cd $SCRATCH_FOLDER
 
 echo "jobstart $(date)";pwd
+echo SLURM_NTASKS
 
-python $SCRATCH/PINN/main.py --2d --dist --epochs 30000 --bs 872356
-# python $SCRATCH/PINN/main.py --upc --2d --dist --bs 8192 --epochs 500
+python $SCRATCH/PINN/main.py --2d --epochs 10000 --bs 160000 -j 0.025 --train_x_step 0.05 0.05
 
 echo "jobend $(date)";pwd