Dec-10-2019, 07:29 AM
i trained model on google-colab, then i saved it with pickle, then downloaded.at last i loaded model with pickle, but it is not working, i mean i should output 2d vector and outputs much more dimentional one for one sample. please help.
with open("model", 'rb') as f:
model=pickle.load(f)
here i am reading model file which is in same directry.
a=torch.ones([10000])
print(a)
print(model.parameters)
print(model(a))
here is error while inputing 10000 dimentional vector
with open("model", 'rb') as f:
model=pickle.load(f)
here i am reading model file which is in same directry.
a=torch.ones([10000])
print(a)
print(model.parameters)
print(model(a))
here is error while inputing 10000 dimentional vector