Pate_alibi 代码阅读记录

Pate_alibi 代码阅读记录

  • 文件夹目录如下
    ├── CONTRIBUTING.md
    ├── LICENSE
    ├── README.md
    ├── aggregate_votes.py
    ├── cifar10
    │   └── cifar10
    ├── lib
    │   ├── init.py
    │   ├── pycache
    │   ├── dataset
    │   ├── fixmatch.py
    │   ├── models
    │   └── pate
    ├── memorization_attack
    │   ├── attack.py
    │   ├── confs
    │   └── confs_10x100
    ├── pate
    │   ├── teacher_0.ckp
    │   └── teacher_0.ckp_ema
    ├── requirements.txt
    ├── train_cifar_alibi.py
    ├── train_student.py
    └── train_teacher.py
  1. 整个代码贯穿着 logging 模块,很有利于调试,是很不错的习惯
  2. 使用 simple_parsing 处理命令行参数(对于后续实验调整参数非常方便)
  3. 整体代码架构也很清晰,将 setting 独立出来