From 8985bbedf3d4204b7f9c2d3120f5ce88a4a12c0e Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Tue, 7 May 2019 19:17:35 +0800 Subject: [PATCH] Add P-DARTS --- lib/nas/genotypes.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/lib/nas/genotypes.py b/lib/nas/genotypes.py index 51b2d60..d2b93d5 100644 --- a/lib/nas/genotypes.py +++ b/lib/nas/genotypes.py @@ -208,12 +208,37 @@ GDAS_FG = Genotype( reduce_concat=range(2, 6) ) +PDARTS = Genotype( + normal=[ + ('skip_connect', 0, 1.0), + ('dil_conv_3x3', 1, 1.0), + ('skip_connect', 0, 1.0), + ('sep_conv_3x3', 1, 1.0), + ('sep_conv_3x3', 1, 1.0), + ('sep_conv_3x3', 3, 1.0), + ('sep_conv_3x3', 0, 1.0), + ('dil_conv_5x5', 4, 1.0)], + normal_concat=range(2, 6), + reduce=[ + ('avg_pool_3x3', 0, 1.0), + ('sep_conv_5x5', 1, 1.0), + ('sep_conv_3x3', 0, 1.0), + ('dil_conv_5x5', 2, 1.0), + ('max_pool_3x3', 0, 1.0), + ('dil_conv_3x3', 1, 1.0), + ('dil_conv_3x3', 1, 1.0), + ('dil_conv_5x5', 3, 1.0)], + reduce_concat=range(2, 6) +) + + model_types = {'DARTS_V1': DARTS_V1, 'DARTS_V2': DARTS_V2, 'NASNet' : NASNet, 'PNASNet' : PNASNet, 'AmoebaNet': AmoebaNet, 'ENASNet' : ENASNet, + 'PDARTS' : PDARTS, 'GDAS_V1' : GDAS_V1, 'GDAS_F1' : GDAS_F1, 'GDAS_GF' : GDAS_GF,