ML.NET - GerjanVlot.BERT-ML.NET

Page / - Изменено Alexandr Fokin в 2021/12/29 13:33

GerjanVlot.BERT-ML.NET

Изменено Alexandr Fokin

в 2021/12/29 13:33

Оглавление

BERT Model for dotnet ML

https://github.com/GerjanVlot/BERT-ML.NET
https://github.com/onnx/models/tree/master/text/machine_comprehension/bert-squad/model

Запуск
1) Скачать модель по второй ссылке. bertsquad-10.onnx
Поместить модель в папку BERT-ML.NET\BERT.WebApi\Model

2) По умолчанию проект использует GPU вычисления и ему требуется CUDA и видеокарта.
Если мы не планируем ее использовать, то необходимо:
2.1) Изменить файл BERT-ML.NET\Microsoft.ML.Models.BERT\Onnx\OnnxModelConfigurator.cs
var pipeline = _mlContext.Transforms
...
gpuDeviceId: null //0
2.2) Nuget.
Удалить зависимость Microsoft.ML.OnnxRuntime.Gpu
Установить зависимость Microsoft.ML.OnnxRuntime
Установить зависимость Microsoft.ML.OnnxRuntime.Managed