PDARTS 项目启动与配置教程

1. 项目目录结构及介绍

PDARTS项目的目录结构如下：

pdarts/
├── LICENSE
├── README.md
├── Table1.png
├── Table2.png
├── genotypes.py
├── model.py
├── model_search.py
├── operations.py
├── pipeline2.jpg
├── stages.png
├── test.py
├── test_imagenet.py
├── train_cifar.py
├── train_imagenet.py
├── train_search.py
├── utils.py
└── visualize.py

• LICENSE：项目的许可文件，描述了项目的使用条款。
• README.md：项目的说明文件，包含了项目的基本信息和如何使用项目。
• Table1.png 和 Table2.png：可能与项目相关的图表。
• genotypes.py：包含了项目使用的基因型定义，这些定义与网络架构搜索相关。
• model.py：定义了项目使用的基础模型。
• model_search.py：实现了模型的搜索算法。
• operations.py：包含了模型构建时使用的操作定义。
• pipeline2.jpg 和 stages.png：可能与项目流程或阶段相关的图像。
• test.py：用于测试CIFAR10数据集上的模型性能。
• test_imagenet.py：用于测试ImageNet数据集上的模型性能。
• train_cifar.py：用于训练CIFAR数据集上的模型。
• train_imagenet.py：用于训练ImageNet数据集上的模型。
• train_search.py：用于执行模型搜索过程。
• utils.py：包含了项目中使用的工具函数。
• visualize.py：可能用于可视化模型架构或搜索过程。

2. 项目的启动文件介绍

项目的启动主要是通过运行train_search.py文件开始的。这个脚本负责执行模型搜索的整个过程。以下是启动模型搜索的基本命令：

python train_search.py \\
--tmp_data_dir /path/to/your/data \\
--save log_path \\
--add_layers 6 \\
--add_layers 12 \\
--dropout_rate 0.1 \\
--dropout_rate 0.4 \\
--dropout_rate 0.7 \\
--note note_of_this_run

在上述命令中，你需要替换/path/to/your/data为你的数据集路径，log_path为你想要保存日志的路径。其他参数可以根据你的需要进行调整。

3. 项目的配置文件介绍

PDARTS项目主要使用命令行参数来进行配置。在train_search.py、train_cifar.py和train_imagenet.py等脚本中，你可以通过添加不同的命令行参数来调整模型的训练和搜索过程。

例如，以下是一些常用的命令行参数：

• --tmp_data_dir：指定数据集的路径。
• --save：指定保存日志和模型的路径。
• --add_layers：指定在搜索过程中添加的层数。
• --dropout_rate：指定dropout的比例，用于正则化。
• --note：为当前的运行添加一个注释，方便追踪。

具体的参数和配置选项可以在各个脚本的文档字符串中找到，或者通过运行脚本时使用-h参数来查看帮助信息。

请注意，项目的配置可能还需要根据你的计算资源和具体需求进行调整，比如调整批大小、学习率等。这些配置通常需要在脚本中进行直接修改或者通过命令行参数进行设置。