run the specific model

2024-07-11 11:48:51 +02:00
parent f46486e21b
commit bb33ca9a68
3 changed files with 512 additions and 1 deletions
--- a/xautodl/models/cell_searchs/genotypes.py
+++ b/xautodl/models/cell_searchs/genotypes.py
@@ -213,6 +213,13 @@ AllConv3x3_CODE = Structure(
        (("nor_conv_3x3", 0), ("nor_conv_3x3", 1), ("nor_conv_3x3", 2)),
    ]  # node-3
 )
+Number_5374 = Structure(
+    [
+        (("nor_conv_3x3", 0),),  # node-1
+        (("nor_conv_1x1", 0), ("nor_conv_3x3", 1)),  # node-2
+        (("skip_connect", 0), ("none", 1), ("nor_conv_3x3", 2)),  # node-3
+    ]
+)

 AllFull_CODE = Structure(
    [
@@ -271,4 +278,5 @@ architectures = {
    "all_c1x1": AllConv1x1_CODE,
    "all_idnt": AllIdentity_CODE,
    "all_full": AllFull_CODE,
+    "5374": Number_5374,
 }