deepnet | Deep neural network implemented with gnumpy/cudamat | Machine Learning library

I want to use some of the parameters of the original deepnet package, so I set up a custom model. I read Caret's documentation (Custom Model), but it doesn't work.

Here is my code for setting up the customized model:

I tried to train my neural network, and then evaluate it's testing accuracy. I am using the code at the bottom of this post to train. The fact is that for other neural networks, I can evaluate the testing accuracy with my code without issue. However, for this neural network (which I constructed correctly according to the description of the neural network paper), I can't evaluate the testing accuracy properly and its giving me the traceback below. So maybe something's wrong in my forward pass?

Here is the training and testing code:

I am trying to train my neural network, which is written in PyTorch, but I got the following traceback because of incorrect dimensions. Got the following traceback

This is a model based on the description on page 12, section B.3 of the paper https://arxiv.org/pdf/1609.04836.pdf