web-dev-qa-db-fra.com

Lorsque j'utilise Google Colaboratory, comment enregistrer des poids et des images dans mon Google Drive?

J'utilise Google Colaboratory, puis je veux enregistrer les images de sortie dans mon lecteur Google Drive ou SSD, HHD, mais son répertoire est "/ contenu". 

import os     
print(os.getcwd())
# "/content"

alors est-il possible de changer de chemin (disque dur, disque dur SSD, googledrive)?

python-3.xgoogle-colaboratory
3
이성령

Pour économiser le poids, vous pouvez exécuter les opérations suivantes après l’entraînement.

saver = tf.train.Saver()
save_path = saver.save(session, "data/dm.ckpt")
print('done saving at',save_path)

Vérifiez l'emplacement où les fichiers ckpt ont été enregistrés.

import os
print( os.getcwd() )
print( os.listdir('data') )

Enfin, téléchargez le fichier!

from google.colab import files
files.download( "data/dm.ckpt.meta" )
4
Nazmus Sakib

Vous devez installer Google Drive sur votre session Colab.

from google.colab import drive
drive.mount('/content/gdrive')

Ensuite, vous pouvez simplement écrire sur Google Drive comme vous le feriez sur un système de fichiers local, comme ceci:

with open('/content/gdrive/My Drive/file.txt', 'w') as f:
  f.write('content')
3
Tadej Magajna

Jetez un coup d’œil à l’exemple sur l’interfaçage avec des fichiers externes . Le flux de travail général consiste à exporter le fichier dans l'environnement en nuage, puis à le télécharger.

Créons le tracé de l'exemple "Hello, Colaboratory" dans un fichier. J'ai copié le bloc-notes sur Google Drive et lancé les commandes suivantes:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(20)
y = [x_i + np.random.randn(1) for x_i in x]
a, b = np.polyfit(x, y, 1)
f = plt.figure()
_ = plt.plot(x, y, 'o', np.arange(20), a*np.arange(20)+b, '-')

f.savefig( "test.png")

Si nous répertorions les fichiers dans l'environnement Google Collaboratory, nous verrons test.png parmi eux:

import os
print( os.getcwd() )
print( os.listdir() )
# /content
# ['datalab', '.local', '.config', '.forever', '.cache', '.rnd', 'test.png', '.ipython']

Tout ce qui reste à faire est de le télécharger sur ma machine locale en utilisant l'exemple que j'ai lié au début de cette réponse:

from google.colab import files
files.download( "test.png" )

Enfin, si vous avez réellement besoin des fichiers sur Google Drive au lieu de votre ordinateur local, vous pouvez utiliser l'API Google Drive pour déplacer les fichiers en conséquence.

P.S. Si vous n'aimez pas écrire des fichiers dans /content, vous pouvez toujours créer un sous-répertoire et os.chdir() dans celui-ci, mais gardez à l'esprit que ce sous-répertoire est toujours local dans votre environnement cloud et vous oblige à télécharger les fichiers comme ci-dessus.

0
Artem Sokolov