Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
api		api
onnxruntime		onnxruntime
.gitignore		.gitignore
README.md		README.md
requirements.txt		requirements.txt
train.ipynb		train.ipynb
train_all.json		train_all.json

Repository files navigation

tasr

程式架構

train.ipynb : 訓練用的jupyter notebook
api/ : api用的程式碼
onnxruntime/ : 將pytorch轉換為onnx格式的程式碼

程式環境

如requirements.txt

執行範例

執行train.ipynb即可完成模型訓練，並將模型導出為ONNX格式
透過設定train.ipynb中的TRAIN_FOLD參數，即可調整訓練使用的FOLD
訓練完0~3號fold之後，便可執行 api/api_onnx_ensemble.py 開啟inference api

About

No description, website, or topics provided.

Report repository

Releases

No releases published

Packages

No packages published

Languages