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

ViT experiments + various fixes #60

Closed mklasby closed 1 year ago

mklasby commented 1 year ago

Includes ViT implementation, additional data augmentation, various rigl_util upgrades, new lock file, hydra config groups for different model/dataset combos for easily switching b/w experiment types