Skip to main content

Deep Learning Optimizer Benchmark Suite

Project description

# DeepOBS - A Deep Learning Optimizer Benchmark Suite

![DeepOBS](docs/deepobs_banner.png “DeepOBS”)

[![Documentation Status](https://readthedocs.org/projects/deepobs/badge/?version=latest)](https://deepobs.readthedocs.io/en/latest/?badge=latest) [![Build Status](https://travis-ci.com/fsschneider/deepobs.svg?branch=master)](https://travis-ci.com/username/projectname) [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)

DeepOBS is a benchmarking suite that drastically simplifies, automates and improves the evaluation of deep learning optimizers.

It can evaluate the performance of new optimizers on a variety of real-world test problems and automatically compare them with realistic baselines.

DeepOBS automates several steps when benchmarking deep learning optimizers:

  • Downloading and preparing data sets.

  • Setting up test problems consisting of contemporary data sets and realistic deep learning architectures.

  • Running the optimizers on multiple test problems and logging relevant metrics.

  • Reporting and visualization the results of the optimizer benchmark.

![DeepOBS Output](docs/deepobs.jpg “DeepOBS_output”)

The code for the current implementation working with TensorFlow can be found on [Github](https://github.com/fsschneider/DeepOBS).

The full documentation is available on readthedocs: https://deepobs.readthedocs.io/

The paper describing DeepOBS has been accepted for ICLR 2019 and can be found here: https://openreview.net/forum?id=rJg6ssC5Y7

We are actively working on a PyTorch version and will be releasing it in the next months. In the meantime, PyTorch users can still use parts of DeepOBS such as the data preprocessing scripts or the visualization features.

## Installation

pip install git+https://github.com/fsschneider/DeepOBS.git

Note, that the installation process can take a while as it will also automatically download all baseline results.

We tested the package with Python 3.6 and TensorFlow version 1.12. Other versions of Python and TensorFlow (>= 1.4.0) might work, and we plan to expand compatibility in the future.

Further tutorials and a suggested protocol for benchmarking deep learning optimizers can be found on https://deepobs.readthedocs.io/

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

deepobs-1.1.0.tar.gz (56.5 kB view details)

Uploaded Source

Built Distribution

deepobs-1.1.0-py3-none-any.whl (139.1 kB view details)

Uploaded Python 3

File details

Details for the file deepobs-1.1.0.tar.gz.

File metadata

  • Download URL: deepobs-1.1.0.tar.gz
  • Upload date:
  • Size: 56.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/1.13.0 pkginfo/1.5.0.1 requests/2.18.4 setuptools/40.8.0 requests-toolbelt/0.9.1 tqdm/4.31.1 CPython/3.6.7

File hashes

Hashes for deepobs-1.1.0.tar.gz
Algorithm Hash digest
SHA256 bf62597843db167ad9c9901b0da79cdb907fabc9a27ed8543e169bc737fa5abf
MD5 3314eebff3b138836bafe5a9d1c06e8a
BLAKE2b-256 c71b7ba753fccc67daf91b3b185d32e01b5b1a2d08840824dd1854f64a7a9cbe

See more details on using hashes here.

File details

Details for the file deepobs-1.1.0-py3-none-any.whl.

File metadata

  • Download URL: deepobs-1.1.0-py3-none-any.whl
  • Upload date:
  • Size: 139.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/1.13.0 pkginfo/1.5.0.1 requests/2.18.4 setuptools/40.8.0 requests-toolbelt/0.9.1 tqdm/4.31.1 CPython/3.6.7

File hashes

Hashes for deepobs-1.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 3bd16463b928651d08311391164810f3caa0fce8276bd8bb20f35cbbdbd79c51
MD5 5f9f73c814a4523cc092cfc291740729
BLAKE2b-256 0f5f227401d17646ea39632ff9d0b501f32ec9ae93cc49ce9237eee4ddc99d5e

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page