From ccfa9373ae784d9bcda19d47973af9b9c15d7906 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Sat, 4 Jul 2020 23:12:57 +0700 Subject: [PATCH] Update to remove "module" attribute check --- 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 d697a06..fd00b8b 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 not in ["module", "process_group", "reducer"]: + if not k.startswith('_') and k not in ["process_group", "reducer"]: setattr(self.ema, k, v)