From 766b9623419df078faf02edf219688c33eb96dc8 Mon Sep 17 00:00:00 2001 From: sorki Date: Fri, 8 Dec 2023 20:04:57 +0100 Subject: [PATCH] README: wrap layer in MNIST example --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index e3e8e3b..33e6ac7 100644 --- a/README.md +++ b/README.md @@ -26,7 +26,8 @@ type MNIST '[ Convolution 1 10 5 5 1 1, Pooling 2 2 2 2, Relu , Convolution 10 16 5 5 1 1, Pooling 2 2 2 2, Reshape, Relu , FullyConnected 256 80, Logit, FullyConnected 80 10, Logit] - '[ 'D2 28 28, 'D3 24 24 10, 'D3 12 12 10, 'D3 12 12 10 + '[ 'D2 28 28 + , 'D3 24 24 10, 'D3 12 12 10 , 'D3 12 12 10 , 'D3 8 8 16, 'D3 4 4 16, 'D1 256, 'D1 256 , 'D1 80, 'D1 80, 'D1 10, 'D1 10]