Merge pull request #153 from wenh06/master

add compatibility for eval_model for multi-GPU, following the advice of @xungeer29

Author: Tianxiaomo

train.py

@@ -417,7 +417,10 @@ def burnin_schedule(i):
             else:
                 eval_model = Yolov4(cfg.pretrained, n_classes=cfg.classes, inference=True)
             # eval_model = Yolov4(yolov4conv137weight=None, n_classes=config.classes, inference=True)
-            eval_model.load_state_dict(model.state_dict())
+            if torch.cuda.device_count() > 1:
+                eval_model.load_state_dict(model.module.state_dict())
+            else:
+                eval_model.load_state_dict(model.state_dict())
             eval_model.to(device)
             evaluator = evaluate(eval_model, val_loader, config, device)
             del eval_model