updates

2019-12-21 19:48:07 -08:00 · 2019-12-21 19:48:07 -08:00 · 5e203d3b1a
commit 5e203d3b1a
parent b7a53957b3
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -446,7 +446,7 @@ if __name__ == '__main__':
        mixed_precision = False

    # scale hyp['obj'] by img_size (evolved at 416)
-    hyp['obj'] *= opt.img_size / 320.
+    hyp['obj'] *= opt.img_size / 416.

    tb_writer = None
    if not opt.evolve:  # Train normally