Hey i need to build a simple mnist neural network using torch but it doesnt seem to work these are the parameters needed and my code parameters: Network Architecture (see im