Unverified 提交 84a8099b authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

上级 df30426c
......@@ -20,13 +20,13 @@ import torch.nn as nn
import torch.nn.functional as F
import torchvision
from utils.general import LOGGER
try:
import thop # for FLOPs computation
except ImportError:
thop = None
LOGGER = logging.getLogger(__name__)
@contextmanager
def torch_distributed_zero_first(local_rank: int):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论