TF version
ilyanelken opened this issue · comments
Hi, is there a Tensor Flow version?
Unfortunately not, we only provide a PyTorch implementation. However, I feel most of the code should be directly translatable into TF2
JJGO / shrinkbench
PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.