From 90575107e7b06d48ef91ffa46a41a55439ebdab1 Mon Sep 17 00:00:00 2001 From: tripleMu Date: Wed, 16 Nov 2022 23:10:15 +0800 Subject: [PATCH] Filter PyTorch 1.13 UserWarnings (#10166) FilterWarning for torch.distributed._all_gather_base Co-authored-by: Glenn Jocher --- utils/torch_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 04a387385..fe934abf1 100644 --- 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')):