The network is not meant to work @my3bikaht, but just to show how I use the model.eval() and infer_flat_size function. So I was not expecting the model to actually work. I just randomly inserted some numbers.
What you are suggesting, can you give an example of this?