Выполнение выводов с использованием pytorch с сохранением весов в .tar-файле с помощью django

Я делаю проект по машинному обучению, в котором мне нужно отобразить прогнозы на веб-странице. Веб-страница создается с помощью 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. пожалуйста, помогите.

Вернуться на верх