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