我正在处理一个Django项目,它应该从一个文件中加载一个保存的Tensorflow模型。我应该在哪里运行这段代码才能有一个全局模型变量?目前,我只是在所有函数之外加载views.py文件中的模型,它正在工作,但我不确定这是否只运行一次,以及它是否是一个干净的解决方案。
发布于 2021-12-22 19:09:41
我建议您开始使用mlops,并尽可能多地预加载所有内容,以便在调用api时,django只负责生成预测。我把我的项目留在这里,我有几个关于ML/Django/Tflite的项目。
模型可以分组在媒体文件中,您甚至可以使用DVC对git/bucket/gdrive进行分组,以避免使用大型文件填充git存储库。通过这样做,您不需要创建全局变量,只需实例化模型,您就可以随时生成预测。
如果你有任何疑问,请毫不犹豫地问我。
链接到我的项目:https://github.com/Nouvellie/django-mlops-docker/blob/main/src/main/apps/mlops/utils/model_loader.py
https://stackoverflow.com/questions/68546031
复制相似问题