How do I set workers in model.predict · Lightning-Universe lightning-flash · Discussion #176 · GitHub

This repository was archived by the owner on Oct 9, 2023. It is now read-only.

aribornstein
Feb 24, 2021

❓ Questions and Help

model.predict is running really slow for image classification. Is there any way to up the number of workers?

Replies: 1 comment

Borda
Mar 15, 2021
Maintainer

I would suggest to use multiprocessing.cpu_count()

0 replies

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment