search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

list out of range #75

Open yulinhuyang opened 4 years ago

yulinhuyang commented 4 years ago

image

yulinhuyang commented 4 years ago

Traceback (most recent call last): File "layer_channel_prune.py", line 273, in compact_model2 = Darknet([compact_model1.hyperparams.copy()] + compact_module_defs, (img_size, img_size)).to(device) File "/data1/visiondection/detection/yolov3-channel-and-layer-pruning-master/yolov3-channel-and-layer-pruning-master/models.py", line 221, in init self.module_list, self.routs, self.hyperparams = create_modules(self.module_defs, img_size, arc) File "/data1/visiondection/detection/yolov3-channel-and-layer-pruning-master/yolov3-channel-and-layer-pruning-master/models.py", line 57, in create_modules filters = sum([output_filters[i + 1 if i > 0 else i] for i in layers]) File "/data1/visiondection/detection/yolov3-channel-and-layer-pruning-master/yolov3-channel-and-layer-pruning-master/models.py", line 57, in filters = sum([output_filters[i + 1 if i > 0 else i] for i in layers]) IndexError: list index out of range