Исходный код вики GerjanVlot.BERT-ML.NET

Редактировал(а) Alexandr Fokin 2021/12/29 13:33

Последние авторы
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