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 deepobs

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.1.tar.gz (56.5 kB view details)

Uploaded Source

Built Distribution

deepobs-1.1.1-py3-none-any.whl (139.6 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: deepobs-1.1.1.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.1.tar.gz
Algorithm Hash digest
SHA256 49c5de356580f0d4d8c5c5a66abc99057464a2eb6226bb61cdd4cf98b3d36f29
MD5 f2b73fe0d3a1e80787884a38132f2ff1
BLAKE2b-256 bedb2a75aa54e2d22779e91605f8091098ca9972978691b3efa303a0f751ed0a

See more details on using hashes here.

File details

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

File metadata

  • Download URL: deepobs-1.1.1-py3-none-any.whl
  • Upload date:
  • Size: 139.6 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.1-py3-none-any.whl
Algorithm Hash digest
SHA256 7b1c7f88407389c9a9e28c7908069f6c516c698a366473c3d2b92371331784b7
MD5 832ed91481bf13f9f6214551aba1dd04
BLAKE2b-256 3f3a48a2d742e4d80dfab38fd0270260508b68c54ac0f2b3c938b7cf189144dd

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