back to half 640 32

tasks/image.py

@@ -129,11 +129,11 @@ async def evaluate_image(request: ImageEvaluationRequest):
     #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
     #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
     #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
-    PATH_TO_MODEL = 'models/best_YOLOv11n_640_half_batch_64.engine'
+    PATH_TO_MODEL = 'models/best_YOLOv11n_640_half_batch_32.engine'
     INFERENCE_ENGINE_TYPE = 'pt'
     INPUT_SIZE = 640
     N_TEST_BATCHES = 2
-    BATCH_SIZE = 64 # Can be adjusted as needed
+    BATCH_SIZE = 32 # Can be adjusted as needed
     print("PATH_TO_MODEL", PATH_TO_MODEL)