Skip to content

Commit

Permalink
Update README to pytorch 0.3.0 and make consistent
Browse files Browse the repository at this point in the history
  • Loading branch information
ottonemo committed Dec 8, 2017
1 parent 950cd9f commit 92a7c59
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

|build| |coverage| |docs|

A scikit-learn compatible neural network library that wraps pytorch.
A scikit-learn compatible neural network library that wraps PyTorch.

.. |build| image:: https://travis-ci.org/dnouri/skorch.svg?branch=master
:alt: Build Status
Expand Down Expand Up @@ -60,7 +60,7 @@ To see a more elaborate example, look `here
X = self.nonlin(self.dense0(X))
X = self.dropout(X)
X = F.relu(self.dense1(X))
X = F.softmax(self.output(X))
X = F.softmax(self.output(X), dim=-1)
return X
Expand Down Expand Up @@ -123,16 +123,16 @@ We recommend to use a virtual environment for this.
From source
~~~~~~~~~~~

If you would like to use the must recent additions to ``skorch`` or
help development, you should install ``skorch`` from source.
If you would like to use the must recent additions to skorch or
help development, you should install skorch from source.

Using conda
^^^^^^^^^^^

You need a working conda installation. Get the correct miniconda for
your system from `here <https://conda.io/miniconda.html>`__.

If you just want to use ``skorch``, use:
If you just want to use skorch, use:

.. code:: bash
Expand Down Expand Up @@ -161,7 +161,7 @@ If you want to help developing, run:
Using pip
^^^^^^^^^

If you just want to use ``skorch``, use:
If you just want to use skorch, use:

.. code:: bash
Expand All @@ -187,12 +187,12 @@ If you want to help developing, run:
py.test # unit tests
pylint skorch # static code checks
pytorch
PyTorch
~~~~~~~

``pytorch`` is not covered by the dependencies, since the ``pytorch``
PyTorch is not covered by the dependencies, since the PyTorch
version you need is dependent on your system. For installation
instructions for ``pytorch``, visit the `pytorch website
instructions for PyTorch, visit the `PyTorch website
<http://pytorch.org/>`__.

In general, this should work (assuming CUDA 9):
Expand Down

0 comments on commit 92a7c59

Please sign in to comment.