From ce7deec440404d17b315768d955313404d70e776 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 11 Aug 2021 17:32:13 +0200
Subject: [PATCH] `int(mlc)` (#4385)
---
train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.py b/train.py
index 23f4971b1..24152f1a1 100644
--- a/train.py
+++ b/train.py
@@ -204,7 +204,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
hyp=hyp, augment=True, cache=opt.cache, rect=opt.rect, rank=RANK,
workers=workers, image_weights=opt.image_weights, quad=opt.quad,
prefix=colorstr('train: '))
- mlc = np.concatenate(dataset.labels, 0)[:, 0].max() # max label class
+ mlc = int(np.concatenate(dataset.labels, 0)[:, 0].max()) # max label class
nb = len(train_loader) # number of batches
assert mlc < nc, f'Label class {mlc} exceeds nc={nc} in {data}. Possible class labels are 0-{nc - 1}'