Unverified 提交 90575107 authored 作者: tripleMu's avatar tripleMu 提交者: GitHub

Filter PyTorch 1.13 UserWarnings (#10166)

FilterWarning for torch.distributed._all_gather_base Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 166b9f2f
...@@ -32,6 +32,7 @@ except ImportError: ...@@ -32,6 +32,7 @@ except ImportError:
# Suppress PyTorch warnings # Suppress PyTorch warnings
warnings.filterwarnings('ignore', message='User provided device_type of \'cuda\', but CUDA is not available. Disabling') warnings.filterwarnings('ignore', message='User provided device_type of \'cuda\', but CUDA is not available. Disabling')
warnings.filterwarnings('ignore', category=UserWarning)
def smart_inference_mode(torch_1_9=check_version(torch.__version__, '1.9.0')): def smart_inference_mode(torch_1_9=check_version(torch.__version__, '1.9.0')):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论