chore: remove unused models

joennlae · Sep 18, 2023 · 6f74a33 · 6f74a33
1 parent 37913f8
commit 6f74a33
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/python/retraining.py b/src/python/retraining.py
@@ -21,7 +21,6 @@
 from models.resnet import resnet18
 from models.resnet9 import ResNet9
 from models.resnet20 import resnet20
-from models.resnet_georg import ResNet
 from halutmatmul.halutmatmul import HalutModuleConfig
 from halutmatmul.model import HalutHelper, get_module_by_name
 from halutmatmul.modules import HalutConv2d, HalutLinear
@@ -71,8 +70,6 @@ def load_model(
         )
         if args.model == "resnet20":
             model = resnet20()
-        elif args.model == "resnet20_georg":
-            model = ResNet("ResNet20")  # type: ignore
         elif args.model == "resnet9":
             model = ResNet9(3, num_classes)  # type: ignore
     else: