logo

111 Commits (527aff4f6975b353f47b3225bbc16ea41b4a81df)
 

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