pytorch_Highway
pytorch_Highway copied to clipboard
kefirski
Highway network implemented in pytorch
Pytorch Highway network
Highway network implemented in PyTorch.
Usage
highway = Highway(input_size, num_layers, f=torch.nn.functional.relu)
# input is [batch_size, input_size] shaped tensor
out = highway(input)
kefirski