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
Our Free Plans just got better! | Auth0 Icon
Our Free Plans just got better! | Auth0

With up to 25k MAUs and unlimited Okta connections, our Free Plan lets you focus on what you do best—building great apps.

You asked, we delivered! Auth0 is excited to expand our Free and Paid plans to include more options so you can focus on building, deploying, and scaling applications without having to worry about your security. Auth0 now, thank yourself later.
Try free now
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