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,