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

runtimeerror:the size of tensor a (208) must match the size of tensor b (416) at non-singleton dimension 3 #148

Open Kuuuo opened 3 years ago

Kuuuo commented 3 years ago

想给模型添加注意力机制，但是在train过程中报这个错误，请问有人遇到过这个问题吗？如何解决呢？

runtimeerror:the size of tensor a (208) must match the size of tensor b (416) at non-singleton dimension 3