certified

architectures/Model.py

@@ -139,7 +139,7 @@ class Model(ABC):
 
 
 
-    # Using the factory patern here
+    # factory
     @staticmethod
     def create(arch, device, size):
         print(f'>> MODEL ARCHITECTURE  >> {arch.name}.')
@@ -151,6 +151,11 @@ class Model(ABC):
                 from architectures.ResNet18 import ResNet18
                 return ResNet18(device, size)
             
+            # ResNet34
+            case Architecture.RESNET34: 
+                from architectures.ResNet34 import ResNet34
+                return ResNet34(device, size)
+            
             # ResNet50
             case Architecture.RESNET50:
                 from architectures.ResNet50 import ResNet50
@@ -190,6 +195,7 @@ from enum import Enum, auto
 class Architecture(Enum):
     RESNET18 = auto()
     RESNET50 = auto()
+    RESNET34 = auto()
     INCEPTION = auto()
     DENSENET121 = auto()
     GOOGLENET = auto()