Как экспортировать настроенную модель SetFit в формат ONNX? ⇐ Python
Как экспортировать настроенную модель SetFit в формат ONNX?
SetFit имеет встроенную функциональность для экспорта в ONNX, но нет реальной документации о том, как это сделать. Что следует передать для параметров setfit_model и inputs в функции export_onnx_setfit_model?
# Загрузить локально сохраненную предварительно обученную модель setfit_model = SetFitModel.from_pretrained("путь/к/моей/модели", local_files_only=True) # Создаем оболочку SetFit onnx onnx_setfit_model = OnnxSetFitModel(model_body=setfit_model.model_body) # Экспорт модели Export_onnx_setfit_model(setfit_model=onnx_setfit_model, inputs=???, output_path="путь/к/выходу")
SetFit имеет встроенную функциональность для экспорта в ONNX, но нет реальной документации о том, как это сделать. Что следует передать для параметров setfit_model и inputs в функции export_onnx_setfit_model?
# Загрузить локально сохраненную предварительно обученную модель setfit_model = SetFitModel.from_pretrained("путь/к/моей/модели", local_files_only=True) # Создаем оболочку SetFit onnx onnx_setfit_model = OnnxSetFitModel(model_body=setfit_model.model_body) # Экспорт модели Export_onnx_setfit_model(setfit_model=onnx_setfit_model, inputs=???, output_path="путь/к/выходу")
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение
-
-
Невозможно экспортировать стандартную модель torchvision ResNet50 в файл ONNX.
Anonymous » » в форуме Python - 0 Ответы
- 38 Просмотры
-
Последнее сообщение Anonymous
-
-
-
Как я могу экспортировать модель Opus-MT-FR-EN Pytorch в один файл ONNX?
Anonymous » » в форуме Python - 0 Ответы
- 9 Просмотры
-
Последнее сообщение Anonymous
-
-
-
Как я могу экспортировать модель Opus-MT-FR-EN Pytorch в один файл ONNX?
Anonymous » » в форуме Python - 0 Ответы
- 8 Просмотры
-
Последнее сообщение Anonymous
-