diff --git a/train.py b/train.py
index 83e0561..1fca64c 100644
--- a/train.py
+++ b/train.py
@@ -287,7 +287,7 @@ def train(hyp):
         scheduler.step()
 
         # mAP
-        ema.update_attr(model, include=['md', 'nc', 'hyp', 'names', 'stride'])
+        ema.update_attr(model, include=['md', 'nc', 'hyp', 'gr', 'names', 'stride'])
         final_epoch = epoch + 1 == epochs
         if not opt.notest or final_epoch:  # Calculate mAP
             results, maps, times = test.test(opt.data,