pub_soft/foldingdiff
3
0
Fork 0
mirror of https://github.com/microsoft/foldingdiff.git synced 2026-06-04 13:30:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reorganize slurm scripts

Browse Source
This commit is contained in:
Kevin Wu
2022-09-12 17:42:46 -07:00
parent 8c58f3ef50
commit 258f702fa3
2 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
scripts/slurm/train_cosine_discard_long.sbatch Normal file
View File

@@ -0,0 +1,19 @@
#!/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
# Change to a directory for model to be output to
cd /home/groups/jamesz/wukevin/projects/protdiff_models/full_angles_discard_long
python /home/groups/jamesz/wukevin/projects/protdiff/bin/train.py /home/groups/jamesz/wukevin/projects/protdiff/config_jsons/cath_full_angles_cosine_discard_long.json

0
scripts/train_slurm.sbatch → scripts/slurm/train_discard_long.sbatch
View File