diff --git a/luxonis_train/config/predefined_models/classification_model.py b/luxonis_train/config/predefined_models/classification_model.py
index 37225207..d5d42367 100644
--- a/luxonis_train/config/predefined_models/classification_model.py
+++ b/luxonis_train/config/predefined_models/classification_model.py
@@ -30,7 +30,7 @@ def get_variant(variant: VariantLiteral) -> ClassificationVariant:
         ),
         "heavy": ClassificationVariant(
             backbone="ResNet",
-            backbone_params={"variant": "101"},
+            backbone_params={"variant": "50"},
         ),
     }