google colaboratory, weight download (export saved models)

后端 未结 7 1719
执念已碎
执念已碎 2021-01-30 14:36

I created a model using Keras library and saved the model as .json and its weights with .h5 extension. How can I download this onto my local machine?

to save the model I

相关标签:
7条回答
  • 2021-01-30 15:35

    Here is a solution that worked for me:

    Setup authentication b/w Google Colab and Your Drive:

    Steps:

    -Paste the code as is below

    -This process will generate two URLs for authentication to complete, where you would have to copy the tokens and paste in the bar provided

    !apt-get install -y -qq software-properties-common python-software-properties module-init-tools
    !add-apt-repository -y ppa:alessandro-strada/ppa 2>&1 > /dev/null
    !apt-get update -qq 2>&1 > /dev/null
    !apt-get -y install -qq google-drive-ocamlfuse fuse
    from google.colab import auth
    auth.authenticate_user()
    from oauth2client.client import GoogleCredentials
    creds = GoogleCredentials.get_application_default()
    import getpass
    !google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret} < /dev/null 2>&1 | grep URL
    vcode = getpass.getpass()
    !echo {vcode} | google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret}
    

    Once this authentication is done, use the following codes to establish the connection:

    !mkdir -p drive
    !google-drive-ocamlfuse drive
    

    Now to see the list of files in your Google Drive:

    !ls drive
    

    To save the Keras model output to Drive, the process is exactly the same as storing in local drive:

    -Run the Keras model as usual

    Once the model is trained say you want to store your model outputs (.h5 and json) into the app folder of your Google Drive:

    model_json = model.to_json()
    with open("drive/app/model.json", "w") as json_file:
        json_file.write(model_json)
    # serialize weights to HDF5
    model.save_weights("drive/app/model_weights.h5")
    print("Saved model to drive")
    

    You will find the files in the respective folder of Google Drive, from where you can download as we can see below:

    0 讨论(0)
提交回复
热议问题