Problem while loading model

1

i trained model on google-colab, then saved it with pickle module, and then i load it on my jupyter notebook.it is not working.look at that picture

image
image1366×768 33.2 KB

here i am printing net parameters and it is obvious that output is 2, if input is 10^4, but it outputs different dimentional vector.please help.

2

You are printing the method model.parameters.
If you want to see all parameters, use print(list(model.parameters())).