updates

2019-04-27 17:52:43 +02:00 · 2019-04-27 17:52:43 +02:00 · 469dede6d1
commit 469dede6d1
parent d25190e15b
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -66,7 +66,7 @@ def train(
    # Configure run
    data_cfg = parse_data_cfg(data_cfg)
    train_path = data_cfg['train']
-    nc = data_cfg['classes']  # number of classes
+    nc = int(data_cfg['classes'])  # number of classes

    # Initialize model
    model = Darknet(cfg, img_size).to(device)