Django后端引用深度学习模型报错“No Model Named xxx”

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

“No Model Named BiLSTM”

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

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

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

你可能感兴趣的:(django,python,神经网络,深度学习,人工智能)