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

Add tests before pushing to Docker Hub (#3924)

Browse source
This commit is contained in:
Glenn Jocher 2023-07-24 23:56:16 +02:00 • committed by GitHub
parent 86b3c001c4
commit 2ee147838a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 71 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/utils/torch_utils.py
View file

@ -17,7 +17,7 @@ import torch.nn.functional as F
import torchvision
from ultralytics.utils import DEFAULT_CFG_DICT, DEFAULT_CFG_KEYS, LOGGER, RANK, __version__
from ultralytics.utils.checks import check_requirements, check_version
from ultralytics.utils.checks import check_version
try:
import thop
@ -54,8 +54,7 @@ def smart_inference_mode():
def get_cpu_info():
"""Return a string with system CPU information, i.e. 'Apple M2'."""
check_requirements('py-cpuinfo')
import cpuinfo # noqa
import cpuinfo # pip install py-cpuinfo
k = 'brand_raw', 'hardware_raw', 'arch_string_raw' # info keys sorted by preference (not all keys always available)
info = cpuinfo.get_cpu_info() # info dict