#!/bin/bash

# CUDA_VISIBLE_DEVICES=0 python3 mnbab_certify.py \
#        --dataset mnist --load-certify-file cert.json \
#        --net cnn_7layer_bn \
#        --cert-weight 1  \
#        --load-model your-model-path-here \
#        --test-eps 0.1 --test-steps 200 \
#        --train-batch 1 --test-batch 20 \
#        --mnbab-config ./MNBAB_configs/mnist_eps0.1.json

CUDA_VISIBLE_DEVICES=1 python3 mnbab_certify.py \
       --dataset mnist --load-certify-file cert.json \
       --net cnn_3layer_bn \
       --load-model ../cert-base-master/test_models/mnist/eps0.3/TAPS_trained/cnn_3layer_bn/init_fast/fast_reg_0.5/last_block_5/model.ckpt \
       --test-eps 0.3 --test-steps 200 \
       --train-batch 1 --test-batch 20 \
       --mnbab-config ./MNBAB_configs/mnist_eps0.3.json

# CUDA_VISIBLE_DEVICES=0 python3 mnbab_certify.py \
#        --dataset cifar10 --load-certify-file cert.json \
#        --net cnn_7layer_bn \
#        --cert-weight 1  \
#        --load-model your-model-path-here \
#        --test-eps 0.00784313725 --test-steps 200 \
#        --train-batch 1 --test-batch 10 \
#        --mnbab-config ./MNBAB_configs/cifar10_small_eps.json

# CUDA_VISIBLE_DEVICES=0 python3 mnbab_certify.py \
#        --dataset cifar10 --load-certify-file cert.json \
#        --net cnn_7layer_bn \
#        --cert-weight 1  \
#        --load-model your-model-path-here \
#        --test-eps 0.03137254901 --test-steps 200 \
#        --train-batch 1 --test-batch 10 \
#        --mnbab-config ./MNBAB_configs/cifar10_large_eps.json
