Сохранение и модель fast.ai v1 ULMFiT. с / на диск

Question

Простой вопрос.

Я проводил эксперименты с ULMFiT

И все же я не вижу способа легко сохранять и загружать модели на / с диска? Существует класс загрузки, доступный классу RNNLearner, который создан в этом примере, но он не задокументирован.

Я пробовал несколько методов, и ни один из них не работает. Есть что-то, чего мне не хватает? Кажется, что нет примеров быстрой модели ai v1, загруженной с диска.

Answer 1

Класс RNNLearner () имеет метод save_encoder(). Например, если ваш RNNLearner() объект называется learn, то вызов метода save_encoder() будет следующим: learn.save_encoder(name='my_model_name')

Кроме того, класс RNNLearner() наследуется от класса Learner () , который сам содержит метод save().

Оба метода save_encoder() и save() имеют аргумент name, который принимает строку, которая будет служить именем, под которым сохраняется ваша модель.