Filter PyTorch 1.13 UserWarnings (#10166)

FilterWarning for torch.distributed._all_gather_base Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-16 23:10:15 +08:00 · 2022-11-16 23:10:15 +08:00 · 90575107e7
parent 166b9f2fa7
commit 90575107e7
1 changed files with 1 additions and 0 deletions
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@ -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')):