Как сохранить тензорную модель потока в Google Colab?

Question

Я тренировал модель в Tensorflow, и это заняло у меня 4 часа. Это модель = Sequential (). Теперь я хотел бы сохранить его, вспомнить в других записных книжках и сравнить. Однако я попытался использовать маринованные огурцы, когда пытался сбросить его на диск Google. Это дает мне следующую ошибку. Что я делаю не так, пожалуйста?

----> 1 pickle.dump (model, fileObject) 2 3 # здесь мы закрываем fileObject 4 fileObject.close () 5 # мы открываем файлдля чтения

Ошибка типа: невозможно выбрать объекты _thread._local

Answer 1

Вы можете сохранить свою модель, позвонив по номеру model.save("my_model.h5"), чтобы сохранить модель и загрузить ее colab для вывода.

Модель keras не поддерживается pickle. Он использует формат двоичных данных HDF5 для сохранения модели в виде двоичных данных на жестком диске.

Вы также можете загрузить свою модель на диск Google и получить к ней доступ в colab, подключив диск Google к текущей записной книжке colab.

как это -

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

with open('/gdrive/My Drive/foo.txt', 'w') as f:
  f.write('Hello Google Drive!')

в следующей ячейке введите это, чтобы просмотреть сохраненный файл

!cat '/gdrive/My Drive/foo.txt'

, пожалуйста, посетите раздел фрагментов кода colab (на левой боковой панели) для получения дополнительной информации. примеры кодов для доступа к гугл-драйву формы colab.