Advertisement

对PT.Darts源代码的详细解析与分析

阅读量:

对博客内容进行详细阐述:

Intuitive Explanation of Differentiable Architecture Search (DARTS)

文章结构概览

  • SearchCNN的构建
      • SearchCNNController
      • SearchCNN
      • SearchCell内部nodes之间DAG的构造
    • 导数计算与参数调整

SearchCNN模型构建

SearchCNNController架构设计

models.search_cnn.SearchCNNController

n_ops = 8, n_nodes = 4

之所以存在i + 2这一表达,是由于需要包含两个输入节点,其中一个为前一阶段的输入节点,另一个则为跨层连接所引入的更早阶段的输入节点。

复制代码
    for i in range(n_nodes):
    self.alpha_normal.append(nn.Parameter(1e-3*torch

全部评论 (0)

还没有任何评论哟~