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