Pruning on other pytorch image models

yoniaflalo / knapsack_pruning

Implementation of knapsack pruning

Apache License 2.0

28 stars 5 forks source link

Open abhishekcodework opened 2 years ago

abhishekcodework commented 2 years ago

Can we utilise train_pruning.py to prune other pytorch models from Ross Wightman repository? For eg. SEResNeXT26t or CoATNet

yoniaflalo commented 2 years ago

Unfortunately, not yet.