fixed layer construction
This commit is contained in:
+1
-1
@@ -114,7 +114,7 @@ def xor():
|
|||||||
params.num_gibbs_samples = 3
|
params.num_gibbs_samples = 3
|
||||||
|
|
||||||
# Create layer
|
# Create layer
|
||||||
layer = Layer("Layer_0", (3, 16), params)
|
layer = Layer("Layer_0", (3, 1, 0, 16), params)
|
||||||
|
|
||||||
# Init weights
|
# Init weights
|
||||||
layer.init(0.01)
|
layer.init(0.01)
|
||||||
|
|||||||
Reference in New Issue
Block a user