This file provides the commands needed to train a regular model and an EDM model and compare the clone accuracy using KnockoffNets attack

# Train a single model
python src/train.py --dataset mnist --epochs 50 --model conv3 --n_models 1 --exp_id conv3 --lr 0.1 --augment

# Train EDM
python src/train_diverse.py --dataset mnist --dataset_ood kmnist --epochs 50 --model conv3 --n_models 5 --exp_id conv3_edm --lr 0.1 --lamb 0.5 --augment

# Train Hash function
python src/train_hash.py --dataset kmnist --epochs 20 --model conv3 --exp_id hash --lr 0.1 --augment

# Test Knockoffnets clone accuracy on Regular model
python src/attack.py --dataset_tar=mnist --dataset_sur=fashion --n_models=1 --model_tar=conv3 --model_sur=conv3 --epochs=50 --exp_id=conv3 --attack=knockoff --hash_mode=dnn --id_hash=hash

# Test Knockoffnets clone accuracy on EDM model
python src/attack.py --dataset_tar=mnist --dataset_sur=fashion --n_models=5 --model_tar=conv3 --model_sur=conv3 --epochs=50 --exp_id=conv3_edm --attack=knockoff --hash_mode=dnn --id_hash=hash


