[P] Torchelie – Pytorch extended with more optimizers, losses, layers, models, transformations, and even training loops
Hi guys,
I’ve been working on Torchelie for something like two or three months now. Notable additions are :
- bitempered loss
- VQ layer from VQ-VAE
- RAdam
- SPADE / Conditional BatchNorm
- Neural Style loss / Deep Dream loss / Feature viz loss
- Spectral image to reproduce distill.pub’s amazing feature viz results
Anyway, here’s the link: https://github.com/Vermeille/Torchelie/ and the documentation is here: https://torchelie.readthedocs.io/
Criticism and bug reports are absolutely welcome.
Thank you for trying it out 🙂
submitted by /u/Vermeille
[link] [comments]