Update linear layers in tests for layers

alperkesen · Aug 21, 2019 · 215db8c · 215db8c
1 parent 66c42ec
commit 215db8c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/layers.jl b/test/layers.jl
@@ -6,7 +6,7 @@ using Knet: relu, gpu
 
     input = rand(1:10)
     output = rand(1:10)
-    layer = AutoML.Layer(input, output)
+    layer = AutoML.LinearLayer(input, output)
 
     batchsize = map(x->2^x, rand(3:10))
     x = atype(randn(input, batchsize))
@@ -26,8 +26,8 @@ end
     hidden = rand(1:10)
     output = rand(1:10)
 
-    layer = AutoML.Layer(input, hidden)
-    layer2 = AutoML.Layer(hidden, output)
+    layer = AutoML.LinearLayer(input, hidden)
+    layer2 = AutoML.LinearLayer(hidden, output)
     chain = AutoML.LinearChain(layer, layer2)
 
     batchsize = map(x->2^x, rand(3:10))
@@ -48,8 +48,8 @@ end
     hidden = rand(1:10)
     output = rand(1:10)
 
-    layer = AutoML.Layer(input, hidden)
-    layer2 = AutoML.Layer(hidden, output)
+    layer = AutoML.LinearLayer(input, hidden)
+    layer2 = AutoML.LinearLayer(hidden, output)
     chain = AutoML.CategoricalChain(layer, layer2)
 
     batchsize = map(x->2^x, rand(3:10))