Closed delfosseaurelien closed 3 years ago
allow the use of torch multi-gpu for inference use of nn.DataParallel or ray to make the inference on multiple batches
allow the use of torch multi-gpu for inference use of nn.DataParallel or ray to make the inference on multiple batches