I am just getting started with machine learning, but I am trying to use the zero shot learning that was just added recently.
This is how I am initializing the classifier:
classifier = pipeline("zero-shot-classification", model='facebook/bart-large-mnli', device=0)
I’ve noticed that this is using only a single CPU core and a single GPU (even on aws instances with multiple GPUs).
Is there a flag that I need to use to tell it to use multiple GPUs? Or is this classifier limited to single GPU and single core performance?