File size: 1,376 Bytes
178d33b | 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 34 35 36 37 38 39 40 41 | # python scripts/ood/react/sweep_osr.py
import os
config = [
[
'osr_cifar6/cifar6_seed1.yml', 'osr_cifar6/cifar6_seed1_ood.yml',
'resnet18_32x32', 'results/checkpoints/osr/cifar6_seed1.ckpt'
],
[
'osr_cifar50/cifar50_seed1.yml', 'osr_cifar50/cifar50_seed1_ood.yml',
'resnet18_32x32', 'results/checkpoints/osr/cifar50_seed1.ckpt'
],
[
'osr_tin20/tin20_seed1.yml', 'osr_tin20/tin20_seed1_ood.yml',
'resnet18_64x64', 'results/checkpoints/osr/tin20_seed1.ckpt'
],
[
'osr_mnist6/mnist6_seed1.yml', 'osr_mnist6/mnist6_seed1_ood.yml',
'lenet', 'results/checkpoints/osr/mnist6_seed1.ckpt'
],
]
for [dataset, ood_dataset, network, pth] in config:
command = (f"PYTHONPATH='.':$PYTHONPATH \
srun -p dsta --mpi=pmi2 --gres=gpu:1 -n1 \
--cpus-per-task=1 --ntasks-per-node=1 \
--kill-on-bad-exit=1 --job-name=openood \
python main.py \
--config configs/datasets/{dataset} \
configs/datasets/{ood_dataset} \
configs/networks/react_net.yml \
configs/pipelines/test/test_osr.yml \
configs/preprocessors/base_preprocessor.yml \
configs/postprocessors/react.yml \
--network.pretrained False \
--network.backbone.name {network} \
--network.backbone.checkpoint {pth} \
--num_workers 8 \
--merge_option merge &")
os.system(command)
|