https://jdhao.github.io/2022/03/18/torch_accelerate_batch_inference/