RuntimeError: matrices expected, got 1D, 2D tensors

Hey K.Frank,
Thanks for the reply.
MLPUpdated from models is same as MLPUpdated. I just wrote it together for your convinence, but sorry for that.

Thanks for the help.
Still I don’t understand the difference between train and test call for model. Can you please eloborate that part?