Skip to main content

A port of the Keras API to PyTorch

Project description

Olympic PyTorch

Olympic implements a Keras-like API for PyTorch.

The goal of Olympic is to combine the joy of Pytorch's dynamic graph execution with the joy of Keras's high level abstractions for training. Concretely, Olympic contains:

  1. The function. This implements a very similar API to Keras's and model.fit_generator methods in a more functional and less object-oriented fashion and spares you the effort of "hand-rolling" your own training loop.
  2. Callback objects that perform functionality common to most deep learning training pipelines such as learning rate scheduling, model checkpointing and csv logging. These integrate into and spare you the effort of writing boilerplate code.
  3. Some helpful utility functions such as common metrics and some convenience layers from Keras that are missing in PyTorch.

Documentation is available here

This package is available on PyPi.

pip install olympic

Project details

Download files

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

Files for olympic, version 0.1.5
Filename, size File type Python version Upload date Hashes
Filename, size olympic-0.1.5-py3-none-any.whl (12.7 kB) File type Wheel Python version py3 Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page