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

通道剪枝错误, get_mask2 未找到.. #24

Closed Byronnar closed 4 years ago

Byronnar commented 4 years ago 
 Epoch   gpu_mem      GIoU       obj       cls     total      soft   targets  img_size

0%| | 0/309 [00:00<?, ?it/s]Traceback (most recent call last): File "train.py", line 534, in train() # train normally File "train.py", line 313, in train idx2mask = get_mask2(model, prune_idx, 0.95) NameError: name 'get_mask2' is not defined 0%| | 0/309 [00:00<?, ?it/s] 请问是不是 命名有问题? 我改成 get_mask之后也报错

tanluren commented 4 years ago

嗯。。。实验代码，已删掉，使用新版本