YOLOX pruning

Job ID: 35594237

Budget: €30 – €250 EUR

Hello.
We are using a YOLOX (medium size) model in our product.
The model is trained with our dataset and we are looking for a person to prune the model.
The model is currently using around 25GFlops per image and we want to lower this to 7GFlops with the lowest possible impact on its accuracy.
We are using Pytorch framework and the environment should use python 3.6, pytorch 1.7.1 and torchvision 0.8.2.
Best regards.
Related categories: Python Machine Learning (ML)