Update model to B6

dnum-mi · Jun 21, 2023 · 63d5209 · 63d5209
1 parent ae1873a
commit 63d5209
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/src/model.py b/backend/src/model.py
@@ -20,8 +20,8 @@
     'revolver',
     'semi_auto_style_militaire_autre']
 
-MODEL_TORCH = Model.efficientnet_b4
-INPUT_SIZE = 380
+MODEL_TORCH = Model.efficientnet_b6
+INPUT_SIZE = 528
 device = torch.device('cpu')