README.md in ruby-dnn-0.14.2 vs README.md in ruby-dnn-0.14.3
- old
+ new
@@ -53,10 +53,10 @@
@l2 = Dense.new(256)
@l3 = Dense.new(10)
end
def call(x)
- x = InputLayer.(x)
+ x = InputLayer.new(784).(x)
x = @l1.(x)
x = ReLU.(x)
x = @l2.(x)
x = ReLU.(x)
x = @l3.(x)