如何配置 Experiment

本指南介绍如何配置 Katib Experiment 以进行神经架构搜索 (NAS)。

阅读本指南之前，请先按照配置超参数 (HP) 调优 Experiment 的指南来了解 NAS 的通用 Experiment 参数。

配置 Experiment

您可以在 Katib Experiment YAML 文件中配置您的 NAS。

该 YAML 文件定义了潜在网络架构的范围、神经网络图的配置、用于确定最佳值的目标指标以及在架构搜索期间使用的搜索算法。

作为参考，您可以使用高效神经架构搜索 (ENAS) 的 YAML 文件。

以下列表描述了 Experiment 的 YAML 文件中特定于 NAS 的参数。

• nasConfig：NAS 的配置。您可以指定要优化的神经网络设计的配置，包括网络中的层数、操作类型等。

  • graphConfig：定义神经网络有向无环图结构的图配置。您可以指定层数、输入层的 input_sizes 和输出层的 output_sizes。

  • operations：您希望为 ML 模型调优的操作范围。对于每个神经网络层，NAS 算法都会选择其中一个操作来构建神经网络。每个操作都包含一组类似于 HP 调优 Experiment 的参数。

您可以在此处找到所有 NAS 示例。

后续步骤

• 了解NAS 算法。