From 100a73b6e96443b32e601ba09ee6594526441726 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 14 Jul 2024 16:12:17 +0200 Subject: [PATCH] Fix `model.save()` method to FP16 (#14418) Co-authored-by: UltralyticsAssistant --- ultralytics/engine/model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ultralytics/engine/model.py b/ultralytics/engine/model.py index 2c1064bf..f8bc495f 100644 --- a/ultralytics/engine/model.py +++ b/ultralytics/engine/model.py @@ -311,11 +311,13 @@ class Model(nn.Module): AssertionError: If the model is not a PyTorch model. """ self._check_is_pytorch_model() + from copy import deepcopy from datetime import datetime from ultralytics import __version__ updates = { + "model": deepcopy(self.model).half() if isinstance(self.model, nn.Module) else self.model, "date": datetime.now().isoformat(), "version": __version__, "license": "AGPL-3.0 License (https://ultralytics.com/license)",