Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
fepegar committed Jul 24, 2019
1 parent 06d18f0 commit d21f7bc
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,16 +7,16 @@ PyTorch implementation of HighRes3DNet from [Li et al., 2017,
*On the Compactness, Efficiency, and Representation of
3D Convolutional Networks: Brain Parcellation as a Pretext Task*][li].

A 2D version (HighRes2DNet) is also available.
A 2D version (`HighRes2DNet`) is also available.

[li]: https://arxiv.org/pdf/1707.01992.pdf

## Installation

### PyTorch hub
### [PyTorch Hub](https://pytorch.org/hub)

If you are using the nightly version of PyTorch, you can import the model
directly from this repository using [PyTorch hub](https://pytorch.org/hub).
directly from this repository using [PyTorch Hub](https://pytorch.org/hub).

```python
>>> import torch
Expand All @@ -30,7 +30,7 @@ directly from this repository using [PyTorch hub](https://pytorch.org/hub).
>>> model = torch.hub.load(repo, model_name, pretrained=True)
```

### `pip`
### [PyPI](https://pypi.org/)

```shell
$ pip install highresnet
Expand Down

0 comments on commit d21f7bc

Please sign in to comment.