HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix model.save() method to FP16 (#14418)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-07-14 16:12:17 +02:00 • committed by GitHub
parent fa06a018d8
commit 100a73b6e9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/model.py
View file

@ -311,11 +311,13 @@ class Model(nn.Module):
AssertionError: If the model is not a PyTorch model. AssertionError: If the model is not a PyTorch model.
""" """
self._check_is_pytorch_model() self._check_is_pytorch_model()
from copy import deepcopy
from datetime import datetime from datetime import datetime
from ultralytics import __version__ from ultralytics import __version__
updates = { updates = {
"model": deepcopy(self.model).half() if isinstance(self.model, nn.Module) else self.model,
"date": datetime.now().isoformat(), "date": datetime.now().isoformat(),
"version": __version__, "version": __version__,
"license": "AGPL-3.0 License (https://ultralytics.com/license)", "license": "AGPL-3.0 License (https://ultralytics.com/license)",