DARTS代码分析（Pytorch）

深度解析DARTS中的Operations（PyTorch实现）

最新推荐文章于 2024-07-06 06:15:00 发布

转载

最新推荐文章于 2024-07-06 06:15:00 发布 · 1.8k 阅读

6 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/yqpy/p/11453074.html

文章标签：

#人工智能 #python

本文详细介绍了DARTS代码中operations.py文件中的各种操作类，包括PoolBN、Identity、FactorizedReduce、DilConv、SepConv、FacConv和Zero，这些操作用于构建搜索空间，涉及池化、恒等连接、特征图尺寸减半、深度可分离卷积等技术。

最近在看DARTS的代码，有一个operations.py的文件，里面是对各类点与点之间操作的方法。

OPS = {
    'none': lambda C, stride, affine: Zero(stride),
    'avg_pool_3x3': lambda C, stride, affine: PoolBN('avg', C, 3, stride, 1, affine=affine),
    'max_pool_3x3': lambda C, stride, affine: PoolBN('max', C, 3, stride, 1, affine=affine),
    'skip_connect': lambda C, stride, affine: \
        Identity() if stride == 1 else FactorizedReduce(C, C, affine=affine),
    'sep_conv_3x3': lambda C, stride, affine: SepConv(C, C, 3, stride, 1, affine=affine),
    'sep_conv_5x5': lambda C, stride, affine: SepConv(C, C, 5, stride, 2, affine=affine),
    'sep_conv_7x7': lambda C, stride, affine: SepConv(C, C, 7, stride, 3, affine=affine),
    'dil_conv_3x3