Python Dill/Pickle - Нет модуля

Хорошо, ребята,

Вот проблема, с которой я борюсь.

  1. У меня есть команда, которая производит модели машинного обучения, используя тензорный поток; они не знают или не очень заботятся о нашем сайте.

  2. У меня есть веб-команда, которая берет вывод команды ML и использует его как apis. Им нужно принять только несколько случайных файлов и заставить их работать.

Поэтому мы решили попробовать выбрать (или добавить) наши классы моделей, а затем загрузить их и восстановить в них переменные веса; здорово это работает.

Что не работает, так это когда мы отправляем класс определения модели на сервер и пытаемся загрузить его с помощью укропа или маринада, мы получаем следующую ошибку:

Файл "/usr/local/lib/python3.5/dist-packages/dill/dill.py", строка 423, в find_class возвращает StockUnpickler.find_class(self, module, name) ImportError: Нет модуля с именем 'RANDOM_MODEL_CLASS_NAME'

Это говорит мне, что маринад / укроп по какой-то причине требует, чтобы класс находился там, где он был или был доступен при загрузке; который побеждает всю цель засолки / укропа в первую очередь для нас.

Любое понимание того, как заставить это работать? Я пробовал pkl.dump и дампов.

Спасибо за любые идеи!

0 ответов

Другие вопросы по тегам