updates

train.py

@@ -121,10 +121,9 @@ def train(
         for i, (imgs, targets, _, _) in enumerate(dataloader):
             imgs = imgs.to(device)
             targets = targets.to(device)
-
             nt = len(targets)
-            if nt == 0:  # if no targets continue
-                continue
+            # if nt == 0:  # if no targets continue
+            #     continue
 
             # Plot images with bounding boxes
             if epoch == 0 and i == 0:

utils/utils.py

@@ -321,7 +321,7 @@ def build_targets(model, targets):
 
         # Class
         tcls.append(c)
-        if nt:
+        if c.shape[0]:
             assert c.max() <= layer.nC, 'Target classes exceed model classes'
 
     return txy, twh, tcls, indices