#194 · Primary category: Deep Learning Frameworks
TorchSSL
A PyTorch-based library for semi-supervised learning (NeurIPS'21)
Project last updated:08/28/23
GitHub Stars
1.4K
Forks
185
Contributors
13
License
MIT
Why we included this project
TorchSSL is the official PyTorch implementation behind FlexMatch, the curriculum pseudo-labeling paper from NeurIPS 2021, and it bundles nine classic semi-supervised learning methods, including PiModel, Mean-Teacher, MixMatch, and FixMatch, under one interface so head-to-head results are easy to compare. The caveat is real: the project is deprecated, and the README points you to its successor, USB, which trains in roughly an eighth of the time and also covers NLP and audio. Treat TorchSSL as a readable historical reference for how these algorithms are structured and for reproducing published numbers, not as a base for new work. That role still earns its keep, because having the internals of pseudo-labeling and consistency-regularization methods sitting side by side in one codebase makes studying them far simpler.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.