ignite
https://github.com/pytorch/ignite
Python
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported3 Subscribers
Add a CodeTriage badge to ignite
Help out
- Issues
- Feat: implement GradMonitor handler for gradient norm tracking and batch skipping
- [Feature] Implement TopK wrapper
- Add MRR (Mean Reciprocal Rank) metric to rec_sys
- [Feature] Implement TopK wrapper class for metrics
- [Feature]: Add Mean Reciprocal Rank (MRR) metric to rec_sys
- [Feature] Implementation of a TrainingHealthMonitor for real-time gradient norm tracking.
- [Feature] Add reset_on_start to EarlyStopping
- Use `Parallel.spawn_kwargs` as kwargs to the backend when appropriate
- Enhancements to EarlyStopping handler
- Check if `test_classification_report.py::test_distrib_nccl_gpu` is passing on the CI with the recent pytorch
- Docs
- Python not yet supported