condensa icon indicating copy to clipboard operation
condensa copied to clipboard

Published 20 hours ago • verified publisher icon NVlabs

Network Thinning Support

Open andyqian2015 opened this issue 5 years ago • 2 comments

thx for your work, I check the library and not find the thin method which mentioned in the article of "A PROGRAMMABLE APPROACH TO MODEL COMPRESSION"

andyqian2015 avatar Nov 12 '19 07:11 andyqian2015

wish to get your reply soon

andyqian2015 avatar Nov 12 '19 07:11 andyqian2015

Hi Andy,

Thank you for bringing this up. We are actively working on adding thinning support but don't have a definite timeline for it yet. As described in the paper, network thinning requires analyzing the DNN DAG to infer inter-node dependencies and we use PyTorch JIT + ONNX to do this. Unfortunately, we've been unable to come up with a reliable implementation that works across PyTorch and ONNX versions. For now, I'll keep this issue open and post any updates as I have them.

srvm avatar Nov 12 '19 21:11 srvm