Skip to content

Commit

Permalink
feat: create train_combined_range_0-5.sh and train_mask_range_1-5.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
dummyindex committed Jan 13, 2024
1 parent 1080299 commit 495304e
Show file tree
Hide file tree
Showing 2 changed files with 94 additions and 0 deletions.
47 changes: 47 additions & 0 deletions notebooks/ViT_scripts/scripts/cluster/train_combined_range_0-5.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
#!/bin/bash

#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
# Slurm Construction Section
#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::

# job name
#SBATCH --job-name=train_csn

#SBATCH --gres=gpu:1
#SBATCH --exclude=g019,g102,g104,g122,g012,g013,g131,g011
#SBAATCH --nodelist=g104
#SBATCH --cpus-per-task=8

#SBATCH --mem=40G

# partition (queue) declaration
#SBATCH --partition=dept_gpu

# number of requested nodes
#SBATCH --nodes=1

# number of tasks
#SBATCH --ntasks=1

# standard output & error

#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
# User Construction Section
#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::


# node
echo
echo $SLURM_JOB_NODELIST
echo

# check GPU
nvidia-smi -L
python=/net/capricorn/home/xing/ken67/.conda/envs/livecell-tracker/bin/python

python train_classify_ViT_classifier_v14_lightning.py\
--batch_size=32\
--start_frame_idx=1 --end_frame_idx=5\
--frame-type combined\
--model_version "frame_0-5-combined"\

47 changes: 47 additions & 0 deletions notebooks/ViT_scripts/scripts/cluster/train_mask_range_1-5.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
#!/bin/bash

#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
# Slurm Construction Section
#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::

# job name
#SBATCH --job-name=train_csn

#SBATCH --gres=gpu:1
#SBATCH --exclude=g019,g102,g104,g122,g012,g013,g131,g011
#SBAATCH --nodelist=g104
#SBATCH --cpus-per-task=8

#SBATCH --mem=40G

# partition (queue) declaration
#SBATCH --partition=dept_gpu

# number of requested nodes
#SBATCH --nodes=1

# number of tasks
#SBATCH --ntasks=1

# standard output & error

#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
# User Construction Section
#::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::


# node
echo
echo $SLURM_JOB_NODELIST
echo

# check GPU
nvidia-smi -L
python=/net/capricorn/home/xing/ken67/.conda/envs/livecell-tracker/bin/python

python train_classify_ViT_classifier_v14_lightning.py\
--batch_size=32\
--start_frame_idx=1 --end_frame_idx=5\
--frame-type mask\
--model_version "frame_1-5-mask"\

0 comments on commit 495304e

Please sign in to comment.