From f83521ab9a63b7728cd562194afa7d9caf66971f Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Sat, 4 Jul 2020 16:50:33 +0700 Subject: [PATCH] Fix save error for multi-gpu --- utils/torch_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index dd2e6e7..d697a06 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -201,5 +201,5 @@ class ModelEMA: def update_attr(self, model): # Update EMA attributes for k, v in model.__dict__.items(): - if not k.startswith('_') and k != 'module': + if not k.startswith('_') and k not in ["module", "process_group", "reducer"]: setattr(self.ema, k, v)