updates

2019-02-19 22:49:47 +01:00
parent 0772ebf7c9
commit 44333ddd9f
1 changed files with 0 additions and 2 deletions
@@ -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)