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

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.