HCA97 / Mosquito-Classifiction

7th place solution of Aicrowd Mosquito Alert Competition
GNU General Public License v3.0
1 stars 1 forks source link

Speed up ineference #5

Closed HCA97 closed 1 year ago

HCA97 commented 1 year ago
HCA97 commented 1 year ago

PyTorch 2.0.1 and Python 3.10 speed up the inference a lot :P (We can run YoloV8-s (640x640) + ViT-L-14 (224x224) in around 1000ms) in old PyTorch version 1.13 I couldn't run YoloV8-s (1024x1024).

All the libraries claimed they were faster but in the end, PyTorch was the fastest :P