ShiftedBronzes / OpenOOD /scripts /ad /cutpaste /mnist_train_osr_cutpaste.sh
AnonymousUser20's picture
Upload 1314 files
178d33b verified
#!/bin/bash
# sh scripts/ad/cutpaste/mnist_train_osr_cutpaste.sh
GPU=1
CPU=1
node=68
jobname=openood
PYTHONPATH='.':$PYTHONPATH \
# srun -p dsta --mpi=pmi2 --gres=gpu:${GPU} -n1 \
# --cpus-per-task=${CPU} --ntasks-per-node=${GPU} \
# --kill-on-bad-exit=1 --job-name=${jobname} -w SG-IDC1-10-51-2-${node} \
python main.py \
--config configs/datasets/osr_mnist6/mnist6_seed1.yml \
configs/datasets/osr_mnist6/mnist6_seed1_ood.yml \
configs/networks/cutpaste.yml \
configs/pipelines/train/train_cutpaste.yml \
configs/preprocessors/cutpaste_preprocessor.yml \
configs/postprocessors/cutpaste.yml \
--network.pretrained False \
--network.backbone.name lenet \
--network.backbone.pretrained True \
--network.backbone.checkpoint 'results/checkpoints/osr/mnist6_seed1.ckpt' \
--num_workers 8 \
--optimizer.num_epochs 100