From 01a73ec08e67479b03376dc24e6a271d6c470db6 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 11 Jul 2020 12:39:27 -0700
Subject: [PATCH] multi-gpu ckpt filesize bug fix #253

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,