Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

New environment variable VERBOSE (#6353)

Browse Source 
New environment variable `VERBOSE`
This commit is contained in:
johnk2hawaii authored and GitHub committed 2022-01-19 14:32:19 -10:00
1 parent 750c42e43e
commit 9708cf56ea
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
utils/general.py
+2 -1
View File
@@ -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():
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: