Commit 5689235f authored by Christian Fuß's avatar Christian Fuß

changed size parameter for OneHot layer to automatically use architecture...

changed size parameter for OneHot layer to automatically use architecture output size, if not specified elsewise.
parent d1b7e652
Pipeline #156329 passed with stages
in 16 minutes and 28 seconds
......@@ -55,6 +55,5 @@ architecture Alexnet(img_height=224, img_width=224, img_channels=3, classes=10){
Concatenate() ->
FullyConnected(units=10) ->
Softmax() ->
OneHot() ->
predictions;
}
\ No newline at end of file
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment