File size: 730 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 |
#! /bin/bash
#SBATCH --nodes=1
#SBATCH -J mednist_train
#SBATCH -c 4
#SBATCH --gres=gpu:1
#SBATCH --time=2:00:00
#SBATCH -p small
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"
# 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"
python -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"
|