Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/kefirski/pytorch_Highway

Highway network implemented in pytorch
https://github.com/kefirski/pytorch_Highway

highway-network python pytorch

Last synced: 29 days ago
JSON representation

Highway network implemented in pytorch

Awesome Lists containing this project

README

        

# Pytorch Highway network

[Highway network](https://arxiv.org/abs/1505.00387) implemented in [PyTorch](http://www.pytorch.org).

![Highway Equation](images/highway.png)

## Usage

```python
highway = Highway(input_size, num_layers, f=torch.nn.functional.relu)

# input is [batch_size, input_size] shaped tensor
out = highway(input)
```