Исходный код вики GerjanVlot.BERT-ML.NET
Редактировал(а) Alexandr Fokin 2021/12/29 13:33
Последние авторы
author | version | line-number | content |
---|---|---|---|
1 | BERT Model for dotnet ML | ||
2 | |||
3 | https://github.com/GerjanVlot/BERT-ML.NET | ||
4 | https://github.com/onnx/models/tree/master/text/machine_comprehension/bert-squad/model | ||
5 | |||
6 | |||
7 | Запуск | ||
8 | 1) Скачать модель по второй ссылке. bertsquad-10.onnx | ||
9 | Поместить модель в папку BERT-ML.NET\BERT.WebApi\Model | ||
10 | |||
11 | 2) По умолчанию проект использует GPU вычисления и ему требуется CUDA и видеокарта. | ||
12 | Если мы не планируем ее использовать, то необходимо: | ||
13 | 2.1) Изменить файл BERT-ML.NET\Microsoft.ML.Models.BERT\Onnx\OnnxModelConfigurator.cs | ||
14 | {{code language="c#"}} | ||
15 | var pipeline = _mlContext.Transforms | ||
16 | ... | ||
17 | gpuDeviceId: null //0 | ||
18 | {{/code}} | ||
19 | 2.2) Nuget. | ||
20 | Удалить зависимость Microsoft.ML.OnnxRuntime.Gpu | ||
21 | Установить зависимость Microsoft.ML.OnnxRuntime | ||
22 | Установить зависимость Microsoft.ML.OnnxRuntime.Managed |