Cont-Steal

This is a Pytorch implementation of Cont-Steal.

Conduct normal attack

   python3 normal_attack.py --model_type --steal --target_dataset

Conduct cont-steal

   python3 contrastive_steal.py --model_type --target_dataset