Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Jul 4, 2020

Commit

9485f0b

unverified ·

2 Parent(s): 5e2429e ccfa937

Merge pull request #296 from NanoCode012/patch-1

Files changed (1) hide show

utils/torch_utils.py CHANGED Viewed

@@ -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':
                 setattr(self.ema, k, v)

     def update_attr(self, model):
         # Update EMA attributes
         for k, v in model.__dict__.items():
+            if not k.startswith('_') and k not in ["process_group", "reducer"]:
                 setattr(self.ema, k, v)