Skip to main content

Hight level training routine for tensorflow.

Project description


Tfignite is a project that stems from ignite. Ignite is a high-level library to help with training neural networks in PyTorch, while Tfignite bares similar design / api and is designed dedicated to tensorflow 2.0.

For the full API reference, read the online documentation. For example usage, check the examples directory.

Why tfignite?

Tfignite separates training/evaluation loop from model/dataset computation graph. This makes a single training/evaluation script highly portable to different project and developers only need to focus on how to build the model and dataset for their tasks. The difference from Keras is that the training/evaluation loop is not part of Model's APIs; instead, developers define model forward pass function, which is then injected into the loop defined by an Engine. Users can also register event handlers in different phases of a training/evaluation loop (for e.g. ITERATION_STARTED).

Apart from the aforementioned separation of model forward pass function and boilerplate loop in Engine (ignite has full credits for this). Tfignite further reduce boilerplate code by defining the Model, Dataset, Callback, ArgumentParser interfaces:

  1. Model: Defines the create_trainer and the create_evaluator function, both of which injects a forward pass function to an Engine and returns it to the training/evaluation script.
  2. Dataset: Defines an unified interface Dataset.create wrapping over
  3. Callback: Defines a interface to group related Engine event handlers in different phases. For example, Checkpointer loads the checkpoint at the beginning of training and stores the checkpoint on EPOCH_COMPLETED.
  4. ArgumentParser: Inherited from argparse.ArgumentParser, the parser pass itself to the Model and Dataset classes for parsing Model-specific and Dataset-specific arguments. This further separates the Model development and the training/evaluation script.


pip install tfignite

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tfignite-0.0.4.tar.gz (10.4 kB view hashes)

Uploaded source

Built Distribution

tfignite-0.0.4-py3-none-any.whl (12.4 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page