Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # name size parameters
- --- -------- ------------------------- ------------------------
- 0 input 1x28x28 0
- 1 conv2d1 (28-(5-1))=24 -> 32x24x24 (5*5*1+1)*32 = 832
- 2 maxpool1 32x12x12 0
- 3 conv2d2 (12-(3-1))=10 -> 32x10x10 (3*3*32+1)*32 = 9'248
- 4 maxpool2 32x5x5 0
- 5 dense 256 (32*5*5+1)*256 = 205'056
- 6 output 10 (256+1)*10 = 2'570
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement