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
$300 Free Credits to Build on Google Cloud Icon
$300 Free Credits to Build on Google Cloud

New customers can spin up VMs, build with AI, and query data at no cost.

Put your $300 in credit toward real workloads, then keep building with free monthly usage for 20+ products. No commitment and no charge until you upgrade.
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