Skip to content

Rostar-github/FSA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Implementation of FSA

Installation

conda create -n fsa python=3.7
conda activate fsa
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
pip install opacus 

Train FSA

Train without differential privacy

python main.py  --epochs=100 \
                --cli_dataset="cifar10" \
                --adv_dataset="cifar10" \
                --level=3 \ 
                --start_sniff=50 \ # start to sniff, after training 50 epochs on split network
                --dp_mode= 
                

Train with differential privacy

python main.py  --epochs=100 \
                --cli_dataset="cifar10" \
                --adv_dataset="cifar10" \
                --level=3 \ 
                --start_sniff=50 \ # start to sniff, after training 50 epochs on split network
                --dp_mode=True \
                --delta=0.00001 \
                --epsilon=0.01

Evaluate FSA

python performance/evaluate.py --level=3 --dataset="cifar10" --batch_size=16 --gen_batches=32 

About

Implementation of FSA

Resources

License

Stars

1 star

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages