yeah i noticed that typo too and i fixed it. But im having trouble determining the linear’s channels? I have written 8192 but its wrong. Im working on the mnist dataset whose images have the following shape: 1,28,28. I edited the network i posted.