From 3c26277ce845f72e4693cc70254f445c38f46f06 Mon Sep 17 00:00:00 2001 From: "DESKTOP-UL3K6JV\\Daniq" <31.10.93@bk.ru> Date: Sun, 16 Jun 2024 17:55:40 +0500 Subject: [PATCH] Model --- classifiers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/classifiers.py b/classifiers.py index d3a6fe6..f190670 100644 --- a/classifiers.py +++ b/classifiers.py @@ -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()