Как сохранить / сохранить сеанс ноутбука Python с объектами TensorFlow? - PullRequest
0 голосов
/ 13 октября 2019

Допустим, я делаю тренинг по модели TensorFlow в блокноте Jupyter / Ipython с большими вычислительными затратами на создание набора данных (tf.Dataset) и его предварительную обработку.

Я хочу сохранить эти накладные вычисления уже так, чтобы даже если я выключил или снова использовал ноутбук с другими параметрами для обучения. Мне не нужно делать создание и предварительную обработку набора данных еще раз.

Я следовал на этот вопрос и сделал

Сохранить сеанс Notebook:

import dill
dill.dump_session('notebook_env.db')

Восстановление сеанса ноутбука:

import dill
dill.load_session('notebook_env.db')

К сожалению, это дало мне ошибку,

RuntimeError: Tensorflow type 21 not convertible to numpy dtype.

Кто-нибудь знает, как сохранить сеанс ноутбука с tf.Dataset (или объекты TensorFlow в целом)

...