From c6dfe88a75a871cddf78c73586e9c1ec6c2e590d Mon Sep 17 00:00:00 2001 From: triple-Mu Date: Tue, 15 Nov 2022 23:41:52 +0800 Subject: [PATCH] FilterWarning for torch.distributed._all_gather_base --- utils/torch_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 04a3873854ee..fe934abf118c 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -32,6 +32,7 @@ # 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')):