TPP icon indicating copy to clipboard operation
TPP copied to clipboard

verified publisher icon MingSun-Tse
→

Metadata

[ICLR'23] Trainability Preserving Neural Pruning (PyTorch)

Results 2 TPP issues
Sort by recently updated
recently updated
newest added

Memory Usage of GPU keeps increasing, OOM error occurs

2 comment

Thanks for your amazing work! However, when I was running the command for ResNets on ImageNet x3.06 using an A100(80G) gpu, the memory usage continues increase from 27G to 79G...

dywu98 avatar dywu98

How I check the pruning mask and the pruning rate(sparsity) of a pruned and trained model?

NKUShaw avatar NKUShaw

← Metadata

30
Stars
2
Forks
Watchers

Owner

verified publisher icon MingSun-Tse

Metadata

[ICLR'23] Trainability Preserving Neural Pruning (PyTorch)

Back