Как загрузить обученную модель TensorFlow в Google Colab? - PullRequest
Новая
       18

Как загрузить обученную модель TensorFlow в Google Colab?

0 голосов
/ 04 мая 2018

Я новичок и изучаю машинное обучение, я создал модель на Google Colab.

Моя цель - использовать эту модель для автономного прогнозирования в приложении для Android. Поэтому мне нужно скачать обученную модель.

Единственное, что я знаю, это то, что мне нужно сохранить мою модель в виде .pb файла, чтобы создать приложение для Android. Я искал вокруг, может быть, есть ответы, но они были слишком короткими для меня, чтобы понять Поэтому нужны очень подробные ответы.

Этот - мой Test.ipynb файл. Может ли кто-нибудь потратить немного времени на обучение этой модели и посмотреть, сможем ли мы загрузить ее на локальный диск.

1 Ответ

0 голосов
/ 04 мая 2018

Так я сохраняю и загружаю файлы моделей из Collab.

  1. Сохранить только обучаемые переменные (Это мой магазин / восстановить fns):

код указан ниже: 

def store(sess_var, model_path):        
    if model_path is not None:
        saver = tf.train.Saver(var_list=tf.trainable_variables())
        save_path = saver.save(sess_var, model_path)
        print("Model saved in path: %s" % save_path)
    else:
        print("Model path is None - Nothing to store")



def restore(sess_var, model_path):
    if model_path is not None:
        if os.path.exists("{}.index".format(model_path)):
            saver = tf.train.Saver(var_list=tf.trainable_variables())
            saver.restore(sess_var, model_path)
            print("Model at %s restored" % model_path)
        else:
            print("Model path does not exist, skipping...")
    else:
        print("Model path is None - Nothing to restore")

  1. Сжатие каталога, в котором вы сохранили вашу модель. Убедитесь, что в нем больше ничего нет: !tar -czvf model.tar.gz models/

  2. Скачать модель:

    from google.colab import files files.download('model.tar.gz')

Поскольку вы сохраняете только обучаемые переменные вместо всего сеанса, размер модели невелик, и поэтому он может загружаться. Обязательно используйте Chrome - я не смог заставить последний фрагмент работать на Firefox.

...