Django部署深度学习项目-1

如何使用pycharm社区版创建django项目?

安装相关依赖库之后:

创建一个空文件夹,在此文件夹路径下使用命令

django-admin startproject mysite

即可成功创建diango项目

 然后使用命令python manage.py runserver启动服务

 

注意:如果不是在本地启动的服务器,而是在其他服务器上启动的Django项目,需要修改:

a. 修改项目的名称为server/setting.py配置文件将 ALLOWED_HOSTS = [] 改为 ALLOWED_HOSTS = ['*']

b. 运行一下命令python manage.py runserver 0.0.0.0:8000 若端口被占用请修改

c. 访问时,使用服务器ip+8000

在实际生产中,我们往往会将训练好的深度学习模型作为Web系统的后端用于构建系统,在这个过程中可能会出现一个问题,例如我们的模型名称为“BiLSTM”,则会提示

“No Model Named BiLSTM”

        但此时你已经将训练好的模型及模型代码放置在指定路径中,为了django的views或者你的predict函数调用,那么这是为什么呢?

        在这个过程中大家一定要注意,训练时不要保存模型结构及参数,即不要通过torch.save(model, 'BiLSTM.pth'),这样就会有上面的问题,即使你已经将模型代码放置在指定路径中。

        这块的解决办法是:只保存模型参数,然后在predict函数或者views函数中首先将模型结构读取进来,例如你读取模型结构后赋值变量model,然后在利用model的load_state_dict方法读取模型的参数,此时你就会发现成功读取了模型及参数,并且可以完成predict操作或views调用。

猜你喜欢

转载自blog.csdn.net/qq_38563206/article/details/126604616
今日推荐