From ed37551c38906329d129590117fc073c5e6e8d5a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 20 Feb 2019 12:55:06 +0100
Subject: [PATCH] updates

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

diff --git a/models.py b/models.py
index 92562a7b..cc61df1d 100755
--- a/models.py
+++ b/models.py
@@ -158,7 +158,7 @@ class YOLOLayer(nn.Module):
 
             tcls = tcls[mask]
             if xy.is_cuda:
-                txy, twth, mask, tcls = txy.cuda(), twh.cuda(), mask.cuda(), tcls.cuda()
+                txy, twh, mask, tcls = txy.cuda(), twh.cuda(), mask.cuda(), tcls.cuda()
 
             # Compute losses
             nT = sum([len(x) for x in targets])  # number of targets