Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Filter PyTorch 1.13 UserWarnings (#10166)

Browse Source 
FilterWarning for torch.distributed._all_gather_base

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
tripleMuandGlenn Jocher authored and GitHub committed 2022-11-16 16:10:15 +01:00
1 parent 166b9f2fa7
commit 90575107e7
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
utils/torch_utils.py
+1
View File
@@ -32,6 +32,7 @@ except ImportError:
# Suppress PyTorch warnings
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')):