在Django项目中将深度学习模型加载到内存中

xienkqul  于 2023-11-20  发布在  Go
关注(0)|答案(1)|浏览(148)

我有一个深度学习模型,我已经在我的django web REST API中添加了它。我希望模型总是加载在内存中。当前,当用户请求时,模型加载在内存中,然后在一段时间后释放。我希望它不断加载在RAM中,以便我的API的响应时间变得更好。我已经在Windows Server 2019上部署了API。

monwx1rj

monwx1rj1#

您可以从apps.py文件加载它。
在你的django应用程序中(让我们称你的应用程序为myapp),你可能有一个名为apps.py的文件,与你的models.pyviews.py等文件处于同一级别。
你可以重写ready()方法。
myapp/apps.py中的示例

from django.apps import AppConfig

loaded_model = None

class MyAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'myapp'

    def ready(self):
        global loaded_model
        loaded_model = <load your model here>

字符串
settings.py中的INSTALLED_APPS设置中更新您的应用程序,如下所示'myapp.apps.MyAppConfig'
然后,您可以像这样在views.py中使用它

from .apps import loaded_model

def my_view(request):
    result = loaded_model.<your_model_method>

相关问题