Merge branch 'main' into 80-separate-large-vocabularies-use-custom-va…

…lidation-from-small-vocabularies-and-subsets-use-codegen
AllenNeuralDynamics · Nov 19, 2024 · dc6683c · dc6683c
2 parents faeca81 + 97464a3
commit dc6683c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/aind_data_schema_models/__init__.py b/src/aind_data_schema_models/__init__.py
@@ -1,3 +1,3 @@
 """Init package"""
 
-__version__ = "0.6.1"
+__version__ = "0.6.2"
diff --git a/src/aind_data_schema_models/system_architecture.py b/src/aind_data_schema_models/system_architecture.py
@@ -38,3 +38,13 @@ class CPUArchitecture(str, Enum):
     ARM64 = "Arm64"
     ARM = "Arm32"
     RISC_V = "RISC-V"
+
+
+class ModelBackbone(str, Enum):
+    """Model backbones"""
+
+    ALEXNET = "AlexNet"
+    CUSTOM = "Custom"
+    RESNET = "ResNet"
+    VGGNET = "VGGNet"
+    UNET = "UNet"