Update torch_utils.py import `LOGGER` (#5483) · 84a8099b75 - yolov5 - 拓恒技术版本管理系统

Update torch_utils.py import `LOGGER` (#5483)

modifyDataloader

Glenn Jocher

GitHub 3 years ago

parent

df30426c03

commit

84a8099b75

No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23

1 changed files with 2 additions and 2 deletions

Unified View

							
								+ 2
								
								- 2
							
utils/torch_utils.pyView File
								
															import torch.nn.functional as F
															
															import torch.nn.functional as F
														
															import torchvision
															
															import torchvision
														
															from utils.general import LOGGER
														
															try:
															
															try:
														
															    import thop  # for FLOPs computation
															
															    import thop  # for FLOPs computation
														
															except ImportError:
															
															except ImportError:
														
															    thop = None
															
															    thop = None
														
															LOGGER = logging.getLogger(__name__)
															
															@contextmanager
															
															@contextmanager
														
															def torch_distributed_zero_first(local_rank: int):
															
															def torch_distributed_zero_first(local_rank: int):