multi-gpu ckpt filesize bug fix #253

5 years ago · 01a73ec08e
parent a586751904
commit 01a73ec08e
1 changed files with 1 additions and 1 deletions
--- 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,