From 9b32d3cb5d80b098f1588483cab13252cf98aa8d Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 2 Apr 2019 19:19:02 +0200
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index 39ad63ad..76df0a84 100644
--- a/train.py
+++ b/train.py
@@ -40,6 +40,8 @@ def train(
 
     # Initialize model
     model = Darknet(cfg, img_size).to(device)
+    # for m in model.modules():
+    #     weights_init_normal(m)  # set weight distributions
 
     # Optimizer
     lr0 = 0.001  # initial learning rate