Anurag27031994
commented
5 years ago USE get_weights AND set_weights TO SAVE AND LOAD MODEL, RESPECTIVELY.
OPEN THIS LINK TO READ THE SAME CODE PROPERLY:
https://drive.google.com/open?id=1xzrqP7ExTmJiZqVt0A_G6AT69EbIjEI9tUDLD1twqj8
##############################################################################
Assuming that this is your model architecture. However, you may use
whatever architecture, you want to (big or small; any).
def mymodel(): inputShape= (28, 28, 3); model= Sequential() model.add(Conv2D(20, 5, padding="same", input_shape=inputShape)) model.add(Activation('relu')) model.add(Flatten()) model.add(Dense(500)) model.add(Activation('relu')) model.add(Dense(2, activation= "softmax")) return model model.fit(....) #paramaters to start training your model
################################################################################ ################################################################################
once your model has been trained, you want to save your model in your PC
use get_weights() command to get your model weights
weigh= model.get_weights()
now, use pickle to save your model weights, instead of .h5
for heavy model architectures, .h5 file is unsupported.
pklfile= "D:/modelweights.pkl"
try:
fpkl= open(pklfile, 'wb') #Python 3
pickle.dump(weigh, fpkl, protocol= pickle.HIGHEST_PROTOCOL)
fpkl.close()
except:
fpkl= open(pklfile, 'w') #Python 2
pickle.dump(weigh, fpkl, protocol= pickle.HIGHEST_PROTOCOL)
fpkl.close()
################################################################################ ################################################################################
in future, you may want to load your model back
use pickle to load model weights
pklfile= "D:/modelweights.pkl" try: f= open(pklfile) #Python 2
weigh= pickle.load(f);
f.close();except:
f= open(pklfile, 'rb') #Python 3
weigh= pickle.load(f);
f.close();restoredmodel= mymodel()
use set_weights to load the modelweights into the model architecture
restoredmodel.set_weights(weigh)
################################################################################ ################################################################################
now, you can do your testing and evaluation- predictions
y_pred= restoredmodel.predict(X)
Running the files unmodified in Python 3.6.4 on windows I get the following trace:
running
pip3 showfor the libraries:Can you tell me what versions you are using? Seems to be a known problem with Keras per: https://github.com/keras-team/keras/issues/6766 but I assume there's a version that works. Thanks!