ChengZi
|
ea58513a8f
|
add relative package import
|
2 years ago |
Jael Gu
|
a00786d445
|
Update parameters
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
ChengZi
|
f46ad5303f
|
train mlm and clm task
|
2 years ago |
Jael Gu
|
a307a3196a
|
Update model list
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
d838ba8357
|
Replace with transformers.onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
6d228847cf
|
Update save_onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
9acb9072e1
|
Fix log
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
1377e6a08e
|
Fix issue due to enable_onnx_checker only available in torch 0.9
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
f0131944b4
|
Add log to test_onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
73fc63b013
|
Update for onnx test
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
908e07a327
|
Add device
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
f5865f2021
|
Update model list for onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
976368836d
|
Update todo
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
c6eabd756e
|
Add list of supported_model_names(format='onnx')
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
a55e58b29b
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
c1351df718
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
516372f339
|
Allow to save model as onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
7082875b8b
|
Debug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
2bc56f2cd6
|
Add supported_model_names
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
71c4b9e93a
|
Debug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
0bd1bcc127
|
Update model list
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
0ee984c935
|
Modify save_model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
ebfc9c74e3
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
fd966f51bd
|
Support save model with torchscript
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
671ed01d78
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
Jael Gu
|
1389d9923c
|
Refactor
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
Jael Gu
|
923b3c6ff4
|
Refactor operator
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |