1

我正在做一个机器学习项目，我需要在网页上显示预测。该网页是使用 Django 构建的。我有预测函数和模型的权重，但是如何在 Django 代码中集成预测函数、模型和权重并进行预测。

我的预测代码

def predicting(model, device, loader):
    model.eval()
    total_preds = torch.Tensor()
    total_labels = torch.Tensor()
    with torch.no_grad():
        for solute_graphs, solvent_graphs, solute_lens, solvent_lens in loader:
            outputs, i_map = model(
            [solute_graphs.to(device), solvent_graphs.to(device), torch.tensor(solute_lens).to(device),
             torch.tensor(solvent_lens).to(device)])   
            print(outputs)
            total_preds = torch.cat((total_preds, outputs.cpu()), 0)
    return total_preds.numpy().flatten()

我已将权重保存在.tar文件中，因此我需要在加载权重进行预测时运行模型。我不知道在哪里保存我的 PyTorch 模型以及使用 Django 进行推理的权重。请帮忙。

4

1 回答 1

2

在 30 分钟内使用 Django 在 Heroku 部署 PyTorch 深度学习分类器

https://www.youtube.com/watch?v=MLk2We1rJPs

在 Django 应用程序中使用 PyTorch：

https://stefanbschneider.github.io/blog/pytorch-django

我希望这能帮助你开始！！！

于 2021-10-17T16:36:11.787 回答