3
3
![ DeepOBS] ( docs/deepobs_banner.png " DeepOBS ")
4
4
5
5
[ ![ PyPI version] ( https://badge.fury.io/py/deepobs.svg )] ( https://badge.fury.io/py/deepobs )
6
- [ ![ Documentation Status] ( https://readthedocs.org/projects/deepobs/badge/?version=latest )] ( https://deepobs.readthedocs.io/en/latest/?badge=latest )
6
+ [ ![ Documentation Status] ( https://readthedocs.org/projects/deepobs/badge/?version=stable )] ( https://deepobs.readthedocs.io/en/latest/?badge=stable )
7
7
[ ![ License: MIT] ( https://img.shields.io/badge/License-MIT-yellow.svg )] ( https://opensource.org/licenses/MIT )
8
8
9
9
@@ -27,6 +27,7 @@ DeepOBS automates several steps when benchmarking deep learning optimizers:
27
27
28
28
The code for the current implementation working with ** TensorFlow** can be found
29
29
on [ Github] ( https://github.com/fsschneider/DeepOBS ) .
30
+ A PyTorch version is currently developed and can be accessed via the pre-release or the develop branch (see News section below).
30
31
31
32
The full documentation is available on readthedocs:
32
33
https://deepobs.readthedocs.io/
@@ -35,10 +36,19 @@ The paper describing DeepOBS has been accepted for ICLR 2019 and can be found
35
36
here:
36
37
https://openreview.net/forum?id=rJg6ssC5Y7
37
38
38
- We are actively working on a ** PyTorch** version and will be releasing it in the
39
- next months. In the meantime, PyTorch users can still use parts of DeepOBS such
40
- as the data preprocessing scripts or the visualization features.
39
+ ** If you find any bugs in DeepOBS, or find it hard to use, please let us know.
40
+ We are always interested in feedback and ways to improve DeepOBS.**
41
41
42
+ ## News
43
+
44
+ We are currently working on a new and improved version of DeepOBS, version 1.2.0.
45
+ It will support ** PyTorch** in addition to TensorFlow, has an easier interface, and
46
+ many bugs ironed out. You can find the latest version of it in [ this branch] ( https://github.com/fsschneider/DeepOBS/tree/develop ) .
47
+
48
+ A [ pre-release] ( https://github.com/fsschneider/DeepOBS/releases/tag/v1.2.0-beta0 ) is available now.
49
+ The full release is expected in a few weeks.
50
+
51
+ Many thanks to [ Aaron Bahde] ( https://github.com/abahde ) for spearheading the developement of DeepOBS 1.2.0.
42
52
43
53
## Installation
44
54
@@ -48,5 +58,16 @@ We tested the package with Python 3.6 and TensorFlow version 1.12. Other
48
58
versions of Python and TensorFlow (>= 1.4.0) might work, and we plan to expand
49
59
compatibility in the future.
50
60
61
+ If you want to create a local and modifiable version of DeepOBS, you can do this directly from this repo via
62
+
63
+ pip install -e git+https://github.com/fsschneider/DeepOBS.git#egg=DeepOBS
64
+
65
+ for the stable version, or
66
+
67
+ pip install -e git+https://github.com/fsschneider/DeepOBS.git@develop#egg=DeepOBS
68
+
69
+ for the latest development version.
70
+
71
+
51
72
Further tutorials and a suggested protocol for benchmarking deep learning
52
73
optimizers can be found on https://deepobs.readthedocs.io/
0 commit comments