好吧,伙计们,
这是我正在努力解决的问题。
我有一个团队使用 tensorflow 制作机器学习模型;他们不知道也不关心我们的网站。
我有一个网络团队,负责获取 ML 团队的输出并将其作为 api 进行操作。他们只需要接受一些随机文件并使其工作。
所以我们决定尝试对模型类进行酸洗(或挖掘),然后加载它们并将可变权重恢复到它们中;太好了,这很有效。
不起作用的是,当我们将模型定义类发送到服务器并尝试使用 dill 或 pickle 加载它时,我们会收到以下错误:
文件“/usr/local/lib/python3.5/dist-packages/dill/dill.py”,第 423 行,在 find_class 返回 StockUnpickler.find_class(self, module, name) ImportError: No module named 'RANDOM_MODEL_CLASS_NAME'
这告诉我泡菜/莳萝由于某种原因需要类在加载时所在的位置或可访问的位置;这首先对我们来说破坏了酸洗/挖掘它的全部目的。
关于如何让它发挥作用的任何见解?我试过 pkl.dump 和转储。
感谢您的任何见解!