Closed wasd52030 closed 5 months ago
ok, I solve the error
when build model, it should be write below
let cnn =
let input = keras.Input(Shape(28, 28, 1))
let modelFlow =
input
|> layers.Conv2D(16, Shape(5, 5), activation = "relu").Apply
|> layers.MaxPooling2D(Shape(2,2)).Apply
|> layers.Conv2D(36, Shape(5, 5), activation = "relu").Apply
|> layers.MaxPooling2D(Shape(2,2)).Apply
|> layers.Flatten().Apply
|> layers.Dense(128, activation = "relu").Apply
let output = modelFlow |> layers.Dense(10, activation = "softmax").Apply
keras.Model(input, output, "CNN")
Description
Hi, I'm currently trying to run a f# program to train a simple CNN module
and here comes a problem :
and here is my code and environment
and similar logic works in C#...
I don't know how to fix this problem.
I'd appreciate it if you could give me any suggestions or comments.
Thank you Have a nice day!