Skip to content

llbbcc/PATEandMEAL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 

Repository files navigation

PATEandMEAL

配置:

python 3.6

torch 1.5.0

torchvision 0.6.0

cuda 10.2

对于our method 通过运行一下命令来得到实验结果:

python main.py --teachers ['vgg19_BN'] --student vgg19_BN --d_lr 1e-3 --fc_out 1 --pool_out avg --loss ce --adv 1 --out_layer [0,1,2,3,4] --out_dims [10000,5000,1000,500,10] --gamma [0.001,0.01,0.05,0.1,1] --eta [1,1,1,1,1] --name vgg_test

其中网络结构可换。运行时注意GPU ID。

对于PATE文件,我已经在原始PATE的基础上进行了修改,切换了数据集与网络结构,运行方式为:

python main.py

原始Torch版本PATE文件网址:https://github.com/kamathhrishi/PATE 可自行下载运行。 Tensorflow版本PATE可直接安装API

Teacher模型下载:https://pan.baidu.com/s/1Ov6UAEMzt2N8IOPK9vG_TQ

提取码:8jr4

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published