Description
Hi there,
I've noticed that the backend device is "cpu"
from hyperimpute.plugins.core.device import DEVICE
I was wondering if there could be a way to specify the DEVICE when GPU is available?
Cheers.
Are you interested in working on this improvement yourself?