updates

2019-04-02 19:19:02 +02:00 · 2019-04-02 19:19:02 +02:00 · 9b32d3cb5d
commit 9b32d3cb5d
parent f527d30ccd
1 changed files with 2 additions and 0 deletions
--- 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