| #!/bin/bash |
| |
|
|
| GPU=1 |
| CPU=1 |
| 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} \ |
| python main.py \ |
| --config configs/datasets/mnist/mnist.yml \ |
| configs/datasets/mnist/mnist_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/mnist_lenet_acc99.30.ckpt' \ |
| --num_workers 8 \ |
| --optimizer.num_epochs 100 \ |
| --merge_option merge |
|
|
