From 469dede6d12be9ad252b7c033f6d926fab146e92 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 27 Apr 2019 17:52:43 +0200
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index aa42f780..8c8e5b15 100644
--- 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)