@Sylvain_Ard - whats the impact on your inference time when you run it on a CPU? For my use case the inference time was shooting up (>2-3mins per image vs msec when done on a GPU) Which didn’t make sense for any practical (real-life) deployment
Did you check the D2go library? -  GitHub - facebookresearch/d2go: D2Go is a toolkit for efficient deep learning 10