Skip to content

Instantly share code, notes, and snippets.

@kiranmaya
Created February 7, 2020 23:07
Show Gist options
  • Save kiranmaya/8fede79ec88a1a89d7a0e3dfd58bc2ff to your computer and use it in GitHub Desktop.
Save kiranmaya/8fede79ec88a1a89d7a0e3dfd58bc2ff to your computer and use it in GitHub Desktop.
Colab python
saving model in gogle drive
from google.colab import drive
drive.mount('/content/gdrive')
model_save_name = 'classifier.pt'
path = F"/content/gdrive/My Drive/{model_save_name}"
torch.save(model.state_dict(), path)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment