From 44333ddd9f630cb055fdfe22fc4904a3ca516c1d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 19 Feb 2019 22:49:47 +0100 Subject: [PATCH] updates --- models.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/models.py b/models.py index 6e8b6253..293b66e1 100755 --- a/models.py +++ b/models.py @@ -101,9 +101,7 @@ class Upsample(nn.Module): class YOLOLayer(nn.Module): - def __init__(self, anchors, nC, img_size, yolo_layer, cfg): - # TODO: img_size from hyperparams in cfg file, NOT from parser. Make dynamic super(YOLOLayer, self).__init__() nA = len(anchors)