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: add class_names support to Precision and Recall metrics
- Add get_sequence_transform to Accuracy for seq2seq models
- Fix state_dict handling for max_iters in Engine
- FSDP Support
- feat(metrics): add Character Error Rate (CER) metric
- feat(handlers): add GradientNormLogger for monitoring gradient norms
- Add GradientNormLogger handler for monitoring training stability
- Remove scikit-learn dependency from `CohenKappa` metric by implementing a native PyTorch version
- Distributed may deadlock in `one_rank_first` when an exception occurs on the target rank
- Refactor SpearmanRankCorrelation to native PyTorch and remove SciPy dependency
- Docs
- Python not yet supported