Improve Performance on zero shot classification

Beginners
1

Hello,

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?

3 Likes