diff --git a/hpvm/test/epoch_dnn/torch_dnn/yolo/model.py b/hpvm/test/epoch_dnn/torch_dnn/yolo/model.py
index c4852a0f3abc44017b3cd87f427594cc06c9ec3f..2a34a2fafde174075d0e913d53d7992e062fc279 100644
--- a/hpvm/test/epoch_dnn/torch_dnn/yolo/model.py
+++ b/hpvm/test/epoch_dnn/torch_dnn/yolo/model.py
@@ -61,11 +61,11 @@ class TinyYoloV2(lnn.module.Darknet):
                 ('8_max',           nn.MaxPool2d(2, 2)),
                 ('9_convbatch',     lnn.layer.Conv2dBatchReLU(128, 256, 3, 1, 1, momentum=momentum)),
                 ('10_max',          nn.MaxPool2d(2, 2)),
-                ('11_convbatch',    lnn.layer.Conv2dBatchReLU(256, 512, 3, 1, 1, momentum=momentum)),
+                ('11_convbatch',    lnn.layer.Conv2dBatchReLU(256, 384, 3, 1, 1, momentum=momentum)),
                 ('12_max',          lnn.layer.PaddedMaxPool2d(2, 1, (0, 1, 0, 1))),
-                ('13_convbatch',    lnn.layer.Conv2dBatchReLU(512, 1024, 3, 1, 1, momentum=momentum)),
-                ('14_convbatch',    lnn.layer.Conv2dBatchReLU(1024, 1024, 3, 1, 1, momentum=momentum)),
-                ('15_conv',         nn.Conv2d(1024, len(self.anchors)*(5+self.num_classes), 1, 1, 0)),
+                ('13_convbatch',    lnn.layer.Conv2dBatchReLU(384, 384, 3, 1, 1, momentum=momentum)),
+                ('14_convbatch',    lnn.layer.Conv2dBatchReLU(384, 384, 3, 1, 1, momentum=momentum)),
+                ('15_conv',         nn.Conv2d(384, len(self.anchors)*(5+self.num_classes), 1, 1, 0)),
             ])
         )
         # fmt: on