logo

111 Commits (main)
 

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