wassname/NALU-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/NALU-pytorch.git synced 2026-06-27 16:00:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

readme fix

Browse Source
This commit is contained in:
Kevin
2018-08-05 00:47:45 -07:00
parent 78f4472f85
commit 002c97f21e
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4 -4
View File
@@ -14,12 +14,12 @@ This is a PyTorch implementation of [Neural Arithmetic Logic Units](https://arxi
from models import *
# single layer modules
NAC(in_features, out_features)
NALU(in_features, out_features)
NeuralAccumulatorCell(in_dim, out_dim)
NeuralArithmeticLogicUnitCell(in_dim, out_dim)
# stacked layers
MultiLayerNAC(num_layers, in_dim, hidden_dim, out_dim)
MultiLayerNALU(num_layers, in_dim, hidden_dim, out_dim)
NAC(num_layers, in_dim, hidden_dim, out_dim)
NALU(num_layers, in_dim, hidden_dim, out_dim)
```
## Experiments