Spaces:

pyesonekyaw
/

Image_Recognition-CZ3004_SC2079_Multidisciplinary_Project-NTU_SG

Running

App Files Files Community

pyesonekyaw commited on Apr 24, 2023

Commit

7f71b29

•

1 Parent(s): 550063f

bug fix

Files changed (1) hide show

models/common.py +1 -1

models/common.py CHANGED Viewed

@@ -315,7 +315,7 @@ class DetectMultiBackend(nn.Module):
         super().__init__()
         w = str(weights[0] if isinstance(weights, list) else weights)
         pt = self._model_type(w)[0]
-        fp16 = True  # FP16
         nhwc = False  # BHWC formats (vs torch BCWH)
         stride = 32  # default stride
         cuda = torch.cuda.is_available() and device.type != 'cpu'  # use CUDA

         super().__init__()
         w = str(weights[0] if isinstance(weights, list) else weights)
         pt = self._model_type(w)[0]
+        fp16 = False  # FP16
         nhwc = False  # BHWC formats (vs torch BCWH)
         stride = 32  # default stride
         cuda = torch.cuda.is_available() and device.type != 'cpu'  # use CUDA