Model

2024-06-16 17:55:40 +05:00
parent 48876a130b
commit 3c26277ce8
1 changed files with 1 additions and 1 deletions
@@ -57,7 +57,7 @@ class YOLOClassify:
    def classify(self, yolo_classifier_model, image_to_classify):
        image_to_classify = image_to_classify.permute(0, 3, 1, 2)

-        if yolo_classifier_model.fp16:
+        if yolo_classifier_model.model.fp16:
            image_to_classify = image_to_classify.half()
        else:
            image_to_classify = image_to_classify.float()