Unverified 提交 9708cf56 authored 作者: johnk2hawaii's avatar johnk2hawaii 提交者: GitHub

New environment variable `VERBOSE` (#6353)

New environment variable `VERBOSE`
上级 750c42e4
...@@ -36,6 +36,7 @@ from utils.metrics import box_iou, fitness ...@@ -36,6 +36,7 @@ from utils.metrics import box_iou, fitness
FILE = Path(__file__).resolve() FILE = Path(__file__).resolve()
ROOT = FILE.parents[1] # YOLOv5 root directory ROOT = FILE.parents[1] # YOLOv5 root directory
NUM_THREADS = min(8, max(1, os.cpu_count() - 1)) # number of YOLOv5 multiprocessing threads NUM_THREADS = min(8, max(1, os.cpu_count() - 1)) # number of YOLOv5 multiprocessing threads
VERBOSE = str(os.getenv('VERBOSE', True)).lower() == 'true' # global verbose mode
torch.set_printoptions(linewidth=320, precision=5, profile='long') torch.set_printoptions(linewidth=320, precision=5, profile='long')
np.set_printoptions(linewidth=320, formatter={'float_kind': '{:11.5g}'.format}) # format short g, %precision=5 np.set_printoptions(linewidth=320, formatter={'float_kind': '{:11.5g}'.format}) # format short g, %precision=5
...@@ -54,7 +55,7 @@ def is_kaggle(): ...@@ -54,7 +55,7 @@ def is_kaggle():
return False return False
def set_logging(name=None, verbose=True): def set_logging(name=None, verbose=VERBOSE):
# Sets level and returns logger # Sets level and returns logger
if is_kaggle(): if is_kaggle():
for h in logging.root.handlers: for h in logging.root.handlers:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论