junjie.jiang
|
6ff206bd86
|
opset_version to 14
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
7 months ago |
junjie.jiang
|
527aff4f69
|
Check gpu is available
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
7 months ago |
Jael Gu
|
c1927bd7cb
|
Avoid auto download given local checkpoint path
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
1 year ago |
Jael Gu
|
aea524a9d3
|
Fix input size assert issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
1 year ago |
Jael Gu
|
db6727933a
|
Allow to pass checkpoint path
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
1 year ago |
junjie.jiang
|
505b0b220d
|
enable accelerate
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
Jael Gu
|
7f25a7beab
|
remove dc2
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
ChengZi
|
dbf64995d0
|
enhance robustness of timm tfms
|
2 years ago |
Jael Gu
|
3a82fb4ebe
|
Fix for list inputs
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
junjie.jiang
|
86b69319fc
|
rm _types
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
Jael Gu
|
20b320a77e
|
Add timm version
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
192bbfde9f
|
Update qps test
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
af2086bdce
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
f6033fcc81
|
Update device
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
4f2ad553b6
|
Update due to change of triton_client
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
158d86d00e
|
Fix onnx export device issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
b543b1d130
|
Set logging level
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
19a4200fc2
|
Update readme
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
24e4837ac7
|
Add benchmark with qps test
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
14e813281f
|
Add transform config
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
junjie.jiang
|
9dcd5723a6
|
delete Image dep
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
Jael Gu
|
9ce147c9a9
|
Update model config with new method
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
21b72075da
|
Debug decorator
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
efb75f26b3
|
Add torch no_grad in inference
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
ChengZi
|
c199ad7bb3
|
add training guide link in readme
|
2 years ago |
Jael Gu
|
87379e3c5a
|
Update README
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
dae5eb0a4a
|
Fix config issue to support Triton
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
725ca7d383
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
8c66702529
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
03c85991e3
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
2c8c051543
|
Update requirements
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
c7881ca7a0
|
Fix post process
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
3f503747f3
|
Fix dim issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
0fcdf6d962
|
Fix op for vit models
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
8da18d800a
|
Fix issue for config
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
768f0068b8
|
Remove accelerate
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
c0bf53ea6e
|
Debug for tritonserve
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
c947c6ea84
|
Debug to support TritonServer
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
a8b3af6f10
|
Support TritonServe
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
f65c1a0683
|
Update run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
fa1e717980
|
Fix run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
45ed56e2ca
|
Fix run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
ea091310ca
|
Fix run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
cbb83dc5fd
|
Fix run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
d99e8aebd5
|
Fix image mode issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
7012383a55
|
Fix image mode issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
c5d6f70768
|
Fix image mode issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
f769e4a73d
|
Fix image mode issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
bd669ec553
|
Fix run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
Jael Gu
|
2a0357f7df
|
Fix run.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |