aboutsummaryrefslogtreecommitdiff
path: root/config/memory_network_mlp_2_momentum.py
diff options
context:
space:
mode:
Diffstat (limited to 'config/memory_network_mlp_2_momentum.py')
-rw-r--r--config/memory_network_mlp_2_momentum.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/config/memory_network_mlp_2_momentum.py b/config/memory_network_mlp_2_momentum.py
index 5a881ce..8a0ac81 100644
--- a/config/memory_network_mlp_2_momentum.py
+++ b/config/memory_network_mlp_2_momentum.py
@@ -1,4 +1,5 @@
from blocks.initialization import IsotropicGaussian, Constant
+from blocks.algorithms import Momentum
from blocks.bricks import Tanh