From 002c97f21ebd067b577e9c218b154cf68e867d31 Mon Sep 17 00:00:00 2001 From: Kevin Date: Sun, 5 Aug 2018 00:47:45 -0700 Subject: [PATCH] readme fix --- README.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 867e4b7..b760d56 100644 --- a/README.md +++ b/README.md @@ -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