High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently. Less code than pure PyTorch while ensuring maximum control and simplicity. Library approach and no program's control inversion. Use ignite where and when you need. Extensible API for metrics, experiment managers, and other components. The cool thing with handlers is that they offer unparalleled flexibility (compared to, for example, callbacks). Handlers can be any function: e.g. lambda, simple function, class method, etc. Thus, we do not require to inherit from an interface and override its abstract methods which could unnecessarily bulk up your code and its complexity. Extremely simple engine and event system. Out-of-the-box metrics to easily evaluate models. Built-in handlers to compose training pipeline, save artifacts and log parameters and metrics.

Features

  • Trigger any handlers at any built-in and custom events
  • Checkpointing, early stopping, profiling
  • Parameter scheduling, learning rate finder, and more
  • Speed up the training on CPUs, GPUs, and TPUs
  • Distributed ready out-of-the-box metrics to easily evaluate models
  • Tensorboard, MLFlow, WandB, Neptune, and more

Project Samples

Project Activity

See All Activity >

Categories

Machine Learning

License

BSD License

Follow PyTorch Ignite

PyTorch Ignite Web Site

Other Useful Business Software
MongoDB Atlas runs apps anywhere Icon
MongoDB Atlas runs apps anywhere

Deploy in 115+ regions with the modern database for every enterprise.

MongoDB Atlas gives you the freedom to build and run modern applications anywhere—across AWS, Azure, and Google Cloud. With global availability in over 115 regions, Atlas lets you deploy close to your users, meet compliance needs, and scale with confidence across any geography.
Start Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of PyTorch Ignite!

Additional Project Details

Operating Systems

Windows

Programming Language

Python

Related Categories

Python Machine Learning Software

Registered

2022-02-16