如何在 Flutter 上使用 Kobert 模型?

how to use kobert model on flutter?

提问人:EASY 提问时间:11/16/2023 最后编辑:molbdniloEASY 更新时间:11/16/2023 访问量:16

问:

我使用此代码自定义了 kobert 模型。

model = BertForSequenceClassification.from_pretrained('skt/kobert-base-v1', num_labels=6)

我用我的自定义训练数据微调了这个“模型”。然后,我用这段代码保存模型。

model.save_pretrained('.')

我有这个文件:“config.json”和“model.safetensors”。

我想在 Flutter 上加载我的自定义 kobertmodel。

我该如何处理这些文件?

Flutter 机器学习 模型

评论


答: 暂无答案