【发布时间】:2021-04-20 00:27:48
【问题描述】:
我正在尝试将 Python ML 应用程序(使用 Streamlit 制作)部署到服务器。这个应用实际上加载了一个我之前训练过的 NN 模型,并使用这个模型进行分类预测。
我遇到的问题是因为 TensorFlow 是一个如此大的包(最新的 tensorflow-cpu 版本至少 150MB),我尝试使用的托管服务(Heroku)一直告诉我我超出了存储限制300MB。
我想知道是否有其他人有类似的问题或知道如何解决/解决此问题?
到目前为止我所做的尝试
- 我已经尝试用 tensorflow-cpu 替换 tensorflow 要求,这确实显着减小了大小,但它仍然太大了 -
- 我还尝试将 tensorflow-cpu 版本降级为 tensorflow-cpu==2.1.0,这终于奏效了,但后来我在 model.load() 上遇到了问题(我认为这可能与我将tf 版本,因为它在本地运行良好)
【问题讨论】:
标签: python tensorflow heroku keras streamlit