ChengZi
|
730c0aaae3
|
add training readme
|
2 years ago |
 Jael Gu
|
1d34b7d14c
|
Update Readme
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
43a21ce0b0
|
Update Readme for save_model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
93ed7f5e5e
|
Fix model name issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
824ecf2154
|
Remove pooler
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
862459b19a
|
Update forward method to support TritonServe
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
082f59f277
|
Change accelerate to comment since not yet supported in latest release
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
ChengZi
|
009ddda7fc
|
remove redundant script.
|
2 years ago |
|
ChengZi
|
d2fb633df2
|
change train self.model to self._model
|
2 years ago |
|
Jael Gu
|
7fa55ba6eb
|
Debug for TritonServer
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
ChengZi
|
45c55a398c
|
add print args
|
2 years ago |
|
Jael Gu
|
5328585568
|
Modify save_model to support tritonserver
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
9c26196077
|
Update op to support both models
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
34b3d3f0b8
|
Update run.py with ann_search.milvus_client
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
88086ae2a8
|
Fix run.py for new op.model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
e6c7f3d8e0
|
Support Towhee TritonServer
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
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 |
