| | #!/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 |
| |
|
