如何将tensorflow模型保存到google drive

import os
from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive

gauth = GoogleAuth()
gauth.credentials = GoogleCredentials.from_json("path/to/client_secrets.json")
drive = GoogleDrive(gauth)

model_folder = drive.CreateFile({'title': 'My TensorFlow Model', 'mimeType': 'application/vnd.google-apps.folder'})
model_folder.Upload()

model_file = drive.CreateFile({'title': 'saved_model.pb', 'parents': [{'id': model_folder['id']}]})
model_file.SetContentFile('path/to/saved_model.pb')
model_file.Upload()

variables_folder = drive.CreateFile({'title': 'variables', 'mimeType': 'application/vnd.google-apps.folder', 'parents': [{'id': model_folder['id']}]})
variables_folder.Upload()

for file in os.listdir('path/to/variables'):
    variable_file = drive.CreateFile({'title': file, 'parents': [{'id': variables_folder['id']}]})
    variable_file.SetContentFile('path/to/variables/' + file)
    variable_file.Upload()

model_folder = drive.ListFile({'q': "'root' in parents and trashed=false"}).GetList()[0]
model_files = drive.ListFile({'q': f"'{model_folder['id']}' in parents and trashed=false"}).GetList()

saved_model_pb = next((file for file in model_files if file['title'] == 'saved_model.pb'), None)
variables_folder = next((file for file in model_files if file['title'] == 'variables' and file['mimeType'] == 'application/vnd.google-apps.folder'), None)
variable_files = drive.ListFile({'q': f"'{variables_folder['id']}' in parents and trashed=false"}).GetList()

for file in model_files:
    file.GetContentFile('path/to/save/' + file['title'])

for file in variable_files:
    file.GetContentFile('path/to/save/' + file['title'])