issues
search
calgaryml
/
condensed-sparsity
[ICLR 2024] Dynamic Sparse Training with Structured Sparsity
https://openreview.net/forum?id=kOBkxFRKTA
MIT License
17
stars
3
forks
source link
Segmentation & updated benchmarks
#66
Closed
mklasby
closed
1 year ago
mklasby
commented
1 year ago
Cocoapi added as a submodule
train_segmentation.py working for maskrcnn
augmentations with torchvision v2.transforms added to manipulate images and associated bbox and masks
benchmarking rewrite
condensed sparisty rewrite in v2, to replace prior efforts
torch.compile used to accerlate code
poetry upgrade to 1.6.1
jax added as dependency
custom torch.inductor backends added to dependencies