karthick965938/save_model.py

Last active May 11, 2020 12:35

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/karthick965938/3fa6e2dd4c7b720d4212a3c99f24f2aa.js"></script>
Save karthick965938/3fa6e2dd4c7b720d4212a3c99f24f2aa to your computer and use it in GitHub Desktop.

Download ZIP

Save your model file on google colab

Raw

save_model.py

	model.save('model_weights.pth')
	#you can save your model using above file

	model = core.Model.load('model_weights.pth', ['alien', 'bat', 'witch'])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment