Torch-Pruning icon indicating copy to clipboard operation
Torch-Pruning copied to clipboard

Published 20 hours ago • verified publisher icon VainF

torch pruning + DETR

Open Faisal-Hajari opened this issue 7 months ago • 1 comments

when using the code with DETR models I find it incompatible. I think the tracing doesn't work well with DETR. Has anyone used it with any implementation of DETR ?

In my case. If I restrict the pruning to the backbone of the model it works fine. but when I pass the entire model is when I start facing issues.

Faisal-Hajari avatar Dec 12 '23 17:12 Faisal-Hajari

how do you prune the part of a model. can you share some codes? when i prune the parf od a model,the error occurs: no forword in your model?

ganliqiang avatar Dec 27 '23 09:12 ganliqiang