search

VainF / Torch-Pruning

[CVPR 2023] DepGraph: Towards Any Structural Pruning
https://arxiv.org/abs/2301.12900
MIT License
2.75k stars 337 forks source link

expected input[8, 77, 80, 80] to have 128 channels, but got 77 channels instead #75

Open naivewx opened 2 years ago

naivewx commented 2 years ago

when I use 'Torch-Pruning/examples/coco_yolo/train_with_pruning.py' to pure yolov5s , will have error'RuntimeError: Given groups=1, weight of size [255, 128, 1, 1], expected input[8, 77, 80, 80] to have 128 channels, but got 77 channels instead' @VainF @JonnyKong @Serjio42 ,could you help check this issue?

YellenYeason commented 2 years ago

put all the files which include Torch-Pruning to your file which include yolov5/train.py.