THU-MIG / torch-model-compression

针对pytorch模型的自动化模型结构分析和修改工具集,包含自动分析模型结构的模型压缩算法库

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

BiSeNet是否已经支持

WEIZHIHONG720 opened this issue · comments

谢谢您的工作! BiSeNet是否已经支持?

呃没测过训练,单纯用torchpruner是可以发现并剪的。