Explorar el Código

Add Python version output (#7814)

modifyDataloader
Glenn Jocher GitHub hace 2 años
padre
commit
3356f2609c
No se encontró ninguna clave conocida en la base de datos para esta firma ID de clave GPG: 4AEE18F83AFDEB23
Se han modificado 2 ficheros con 3 adiciones y 3 borrados
  1. +1
    -1
      utils/general.py
  2. +2
    -2
      utils/torch_utils.py

+ 1
- 1
utils/general.py Ver fichero

@@ -256,7 +256,7 @@ def file_age(path=__file__):
return dt.days # + dt.seconds / 86400 # fractional days


def file_update_date(path=__file__):
def file_date(path=__file__):
# Return human-readable file modification date, i.e. '2021-3-26'
t = datetime.fromtimestamp(Path(path).stat().st_mtime)
return f'{t.year}-{t.month}-{t.day}'

+ 2
- 2
utils/torch_utils.py Ver fichero

@@ -18,7 +18,7 @@ import torch.distributed as dist
import torch.nn as nn
import torch.nn.functional as F

from utils.general import LOGGER, file_update_date, git_describe
from utils.general import LOGGER, file_date, git_describe

try:
import thop # for FLOPs computation
@@ -51,7 +51,7 @@ def device_count():

def select_device(device='', batch_size=0, newline=True):
# device = 'cpu' or '0' or '0,1,2,3'
s = f'YOLOv5 🚀 {git_describe() or file_update_date()} torch {torch.__version__} ' # string
s = f'YOLOv5 🚀 {git_describe() or file_date()} Python-{platform.python_version()} torch-{torch.__version__} '
device = str(device).strip().lower().replace('cuda:', '') # to string, 'cuda:0' to '0'
cpu = device == 'cpu'
if cpu:

Cargando…
Cancelar
Guardar