mirror of
https://github.com/microsoft/foldingdiff.git
synced 2026-06-04 13:30:33 +08:00
sherlock AR baseline script
This commit is contained in:
17
scripts/slurm/train_ar_baseline.sbatch
Normal file
17
scripts/slurm/train_ar_baseline.sbatch
Normal file
@@ -0,0 +1,17 @@
|
||||
#!/bin/bash
|
||||
#SBATCH --partition=jamesz
|
||||
#SBATCH --job-name=training # Job name
|
||||
#SBATCH --mail-type=FAIL # Mail events (NONE, BEGIN, END, FAIL, ALL)
|
||||
#SBATCH --mail-user=wukevin@stanford.edu # Where to send mail
|
||||
#SBATCH --nodes=1 # Run all processes on a single node
|
||||
#SBATCH --ntasks=20 # Number of processes
|
||||
#SBATCH --mem=40gb # Job memory request
|
||||
#SBATCH -G 2
|
||||
#SBATCH -C GPU_SKU:RTX_2080Ti
|
||||
#SBATCH --time=6-23:59:59 # Time limit
|
||||
#SBATCH --output=training_%j.log # Standard output and error log
|
||||
. /home/groups/jamesz/miniconda3/etc/profile.d/conda.sh
|
||||
. ~/.bashrc
|
||||
# Activate the appropriate conda environment
|
||||
conda activate /home/groups/jamesz/wukevin/envs/protdiff
|
||||
python /home/groups/jamesz/wukevin/projects/protdiff/bin/train_autoregressive.py
|
||||
Reference in New Issue
Block a user