pytorch模型剪枝學習筆記

pytorch代碼倉庫 pytorch在19年11月份的時候合入了這部分剪枝的代碼。pytorch提供一些直接可用的api,用戶只需要傳入需要剪枝的module實例和需要剪枝的參數名字,系統自動幫助完成剪枝操作,看起來接口挺簡單。比如 def random_structured(module, name, amount, dim) pytorch支持的幾種類型的剪枝策略: 詳細分析 pytorch
相關文章
相關標籤/搜索