Update model to B7

dnum-mi · Jun 21, 2023 · f9a5d0a · f9a5d0a
1 parent 24bc388
commit f9a5d0a
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_b6
-INPUT_SIZE = 528
+MODEL_TORCH = Model.efficientnet_b7
+INPUT_SIZE = 600
 device = torch.device('cpu')