Skip to content

Commit

Permalink
Update readme, add setup.cfg and bump version (pytorch#188)
Browse files Browse the repository at this point in the history
* Update readme, add setup.cfg and bump version

* Update docs URL

* Update README and add side-by-side compare image

* Replace image with dark theme

* Update package name: ignite -> pytorch-ignite
  • Loading branch information
vfdev-5 authored Jun 15, 2018
1 parent 022f04a commit a146516
Show file tree
Hide file tree
Showing 5 changed files with 34 additions and 5 deletions.
30 changes: 27 additions & 3 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -10,22 +10,46 @@ Ignite

Ignite is a high-level library to help with training neural networks in PyTorch.

*Note: Ignite is currently in alpha, and the API is subject to change. A first release coming soon, where we will stabilize the API. Stay tuned.*
- ignite helps you write compact but full-featured training loops in a few lines of code
- you get a training loop with metrics, early-stopping, model checkpointing and other features without the boilerplate

Below we show a side-by-side comparison of using pure pytorch and using ignite to create a training loop
to train and validate your model with occasional checkpointing:

.. image:: assets/ignite_vs_bare_pytorch.png

As you can see, the code is more concise and readable with ignite. Furthermore, adding additional metrics, or
things like early stopping is a breeze in ignite, but can start to rapidly increase the complexity of
your code when "rolling your own" training loop.


Installation
============

From pip:

.. code:: bash
pip install pytorch-ignite
From conda:

.. code:: bash
conda install pytorch-ignite -c pytorch
From source:

.. code:: bash
python setup.py install
*Anaconda & pip coming soon*
Documentation
=============
API documentation and an overview of the library coming soon.
API documentation and an overview of the library can be found `here <https://pytorch.org/ignite/index.html>`_.

Examples
========
Expand Down
Binary file added assets/ignite_vs_bare_pytorch.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
2 changes: 1 addition & 1 deletion ignite/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import ignite.engine
import ignite.handlers

__version__ = '0.1.0a1'
__version__ = '0.1.0'
5 changes: 5 additions & 0 deletions setup.cfg
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
[bdist_wheel]
universal=1

[metadata]
license_file = LICENSE
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def find_version(*file_paths):

setup(
# Metadata
name='ignite',
name='pytorch-ignite',
version=VERSION,
author='PyTorch Core Team',
author_email='soumith@pytorch.org',
Expand Down

0 comments on commit a146516

Please sign in to comment.