From b72fb74ad05b40c560d540280de5a5a1bcbf0ace Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 17 Aug 2019 14:15:27 +0200
Subject: [PATCH] updates

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

diff --git a/models.py b/models.py
index 119aaf4b..eb2fdcea 100755
--- a/models.py
+++ b/models.py
@@ -96,7 +96,7 @@ class YOLOLayer(nn.Module):
     def __init__(self, anchors, nc, img_size, yolo_index):
         super(YOLOLayer, self).__init__()
 
-        self.anchors = torch.from_numpy(anchors)
+        self.anchors = torch.Tensor(anchors)
         self.na = len(anchors)  # number of anchors (3)
         self.nc = nc  # number of classes (80)
         self.nx = 0  # initialize number of x gridpoints