Skip to content

Latest commit

 

History

History
36 lines (25 loc) · 930 Bytes

README.md

File metadata and controls

36 lines (25 loc) · 930 Bytes

The code is based on CMI and OoD distillation is based on single-img-extrapolating.

CMI

  • Attack
wandb sweep sweeps/cmi_cifar10_poi.yml
  • Defense
wandb sweep sweeps/cmi_cifar10.yml

Customization

Add dataset: Edit get_dataset and NORMALIZE_DICT in cmi/registry.py.

Add model: Edit MODEL_DICT in cmi/registry.py to add model architecture. To set pre-trained model files, edit get_pretrained_path in cmi/.

OoD

The OoD dataset include OoD patches generated from one image. Details follow single-img-explorating.

  • Attack
wandb sweep sweeps/ood_cifar10_poi.yml
  • Defense
wandb sweep sweeps/ood_cifar10.yml