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