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

Add datetime to exported model metadata (#3694)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Yonghye Kwon 2023-07-13 02:05:06 +09:00 • committed by GitHub
parent 395cc47c53
commit cfcae388f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/engine/exporter.py
View file

@ -55,6 +55,7 @@ import subprocess
import time import time
import warnings import warnings
from copy import deepcopy from copy import deepcopy
from datetime import datetime
from pathlib import Path from pathlib import Path
import torch import torch
@ -219,6 +220,7 @@ class Exporter:
'description': description, 'description': description,
'author': 'Ultralytics', 'author': 'Ultralytics',
'license': 'AGPL-3.0 https://ultralytics.com/license', 'license': 'AGPL-3.0 https://ultralytics.com/license',
'date': datetime.now().isoformat(),
'version': __version__, 'version': __version__,
'stride': int(max(model.stride)), 'stride': int(max(model.stride)),
'task': model.task, 'task': model.task,