File size: 929 Bytes
57decc6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
#! /bin/bash
#SBATCH --nodes=1
#SBATCH -J mednist_train
#SBATCH -c 4
#SBATCH --gres=gpu:2
#SBATCH --time=2:00:00
#SBATCH -p big
set -v
# change this if run submitted from a different directory
export BUNDLE="$(pwd)/.."
# change this to load a checkpoint instead of started from scratch
CKPT=none
CONFIG="'$BUNDLE/configs/train.yaml', '$BUNDLE/configs/train_multigpu.yaml'"
# change this to point to where MedNIST is located
DATASET="$(pwd)"
# it's useful to include the configuration in the log file
cat "$BUNDLE/configs/train.yaml"
cat "$BUNDLE/configs/train_multigpu.yaml"
# remember to change arguments to match how many nodes and GPUs you have
torchrun --standalone --nnodes=1 --nproc_per_node=2 -m monai.bundle run training \
--meta_file "$BUNDLE/configs/metadata.json" \
--config_file "$CONFIG" \
--logging_file "$BUNDLE/configs/logging.conf" \
--bundle_root "$BUNDLE" \
--dataset_dir "$DATASET"
|