ChengZi
|
44fac86041
|
lazy import evaluate and datasets for avoiding potential error.
|
2 years ago |
 Jael Gu
|
0f9661823b
|
Update benchmark/run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
4ea7a7b9ec
|
Update benchmark/run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
b883739db2
|
Update benchmark/run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
2a39a1697d
|
Update save onnx in evaluation
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
2cf6fabd69
|
Update evaluation method
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
0592e151a0
|
Resolve import issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
ChengZi
|
9345e3bfd3
|
lazy import Trainer for avoiding potential error.
|
2 years ago |
|
Jael Gu
|
4f343f9441
|
Update evaluate with onnx path
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
36653e1a14
|
Update onnx test
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
870c57ecbd
|
Update README
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
0d30fc46a7
|
Add model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
ChengZi
|
ea58513a8f
|
add relative package import
|
2 years ago |
|
Jael Gu
|
98d84e87f9
|
Add evaluation
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
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
|
5351a1cbd5
|
Update README
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
00e0c3bd4f
|
Update README
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
a307a3196a
|
Update model list
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
59f19c7f17
|
Add test onnx
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
|
1da282ff9c
|
Update requirements
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>
|
3 years ago |
|
Jael Gu
|
9dda6285eb
|
Update test scripts
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
c6eabd756e
|
Add list of supported_model_names(format='onnx')
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
a55e58b29b
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
e051d859f1
|
update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
d2f54a94c0
|
update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
c1351df718
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
516372f339
|
Allow to save model as onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
ChengZi
|
ef281cdf08
|
add requirement
|
3 years ago |
|
Jael Gu
|
7082875b8b
|
Debug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
2bc56f2cd6
|
Add supported_model_names
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
98212f64c5
|
Add results
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
71c4b9e93a
|
Debug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
0bd1bcc127
|
Update model list
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
0ee984c935
|
Modify save_model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
ebfc9c74e3
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
fd966f51bd
|
Support save model with torchscript
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
LocoRichard
|
6fbe4054ab
|
[DCO] Refine Readme
Signed-off-by: LocoRichard <lichen.wang@zilliz.com>
|
3 years ago |
|
Jael Gu
|
8f13f68398
|
Update readme
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
6c0ff9995a
|
Update readme
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
b8627b693a
|
Update readme
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
