Skip to content

Instantly share code, notes, and snippets.

@karthick965938
Last active May 11, 2020 12:35
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save karthick965938/3fa6e2dd4c7b720d4212a3c99f24f2aa to your computer and use it in GitHub Desktop.
Save your model file on google colab
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