Architecture for pruning methods analysis using pytorch prune module - View it on GitHub