run_adaptive_imia.py

"""
This file is the script to run the IMIA framework (adaptive setting)
"""

import os
import argparse
from utils.loader import load_config

parser = argparse.ArgumentParser()

######### env configuration ########
parser.add_argument("--cuda", "-c", default=0, type=int)
parser.add_argument("--n_imitate_shadows", default=10, type=int)
parser.add_argument("--dataset", default="mnist", type=str)

args = parser.parse_args()
load_config(args)
print(args)

for shadow_id in range(args.n_imitate_shadows):
    if shadow_id % args.n_gpus == args.cuda:
        os.environ["CUDA_VISIBLE_DEVICES"] = str(shadow_id % 4)
        print(f"Training model {shadow_id}/{args.n_imitate_shadows} ...")
        cmd = f"python imitate_adaptive_imia.py --dataset {args.dataset} --shadow_id {shadow_id} --data_dir {args.data_dir} --savedir {args.savedir} --seed {args.seed} --model_type {args.model_type} --n_imitate_shadows {args.n_imitate_shadows} --margin_weight {args.margin_weight} --warmup_epochs {args.warmup_epochs} --temperature {args.temperature} --n_queries {args.n_queries}"
        os.system(cmd)