DARTS-PT 项目使用教程

DARTS-PT 项目使用教程

darts-pt [ICLR2021 Outstanding Paper] Rethinking Architecture Selection in Differentiable NAS 项目地址: https://gitcode.com/gh_mirrors/da/darts-pt

1. 项目目录结构及介绍

darts-pt/
├── attacker/
├── exp_scripts/
├── nasbench201/
├── sota/
│   └── cnn/
├── .gitignore
├── LICENSE
├── README.md
└── requirements.txt

• attacker/: 包含与攻击相关的代码文件。
• exp_scripts/: 包含实验脚本，用于运行实验。
• nasbench201/: 包含与 NAS-Bench-201 相关的代码文件。
• sota/cnn/: 包含与 SOTA（State-of-the-Art）CNN 模型相关的代码文件。
• .gitignore: Git 忽略文件，指定哪些文件或目录不需要被 Git 跟踪。
• LICENSE: 项目许可证文件，本项目使用 Apache-2.0 许可证。
• README.md: 项目说明文件，包含项目的概述、安装和使用说明。
• requirements.txt: 项目依赖文件，列出了项目运行所需的 Python 包。

2. 项目启动文件介绍

项目的主要启动文件位于 exp_scripts/ 目录下，用于运行不同的实验。以下是一些关键的启动脚本：

• darts-201.sh: 用于在 NAS-Bench-201 上训练超网络。
• darts-proj-201.sh: 用于在 NAS-Bench-201 上进行架构选择（投影）。
• blank-201.sh: 用于在 NAS-Bench-201 上运行 Fix-alpha 版本的实验。
• darts-sota.sh: 用于在 SOTA 数据集上训练超网络。
• darts-proj-sota.sh: 用于在 SOTA 数据集上进行架构选择（投影）。

3. 项目的配置文件介绍

项目的配置主要通过命令行参数进行设置，以下是一些常用的配置参数：

• --resume_epoch: 指定从哪个 epoch 恢复训练。
• --resume_expid: 指定从哪个实验 ID 恢复训练。
• --space: 指定搜索空间（如 s1, s2, s3, s4）。
• --dataset: 指定数据集（如 cifar10, cifar100, svhn）。

例如，运行 darts-201.sh 脚本时，可以通过以下命令指定配置：

bash darts-201.sh --space s3 --dataset cifar10 --resume_epoch 100 --resume_expid search-darts-201-1

以上命令将在 CIFAR-10 数据集上，使用 s3 搜索空间，从第 100 个 epoch 恢复训练，并使用实验 ID search-darts-201-1 进行恢复。

darts-pt [ICLR2021 Outstanding Paper] Rethinking Architecture Selection in Differentiable NAS 项目地址: https://gitcode.com/gh_mirrors/da/darts-pt