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?
Description
Hi there,
I've noticed that the backend device is "cpu"
from hyperimpute.plugins.core.device import DEVICEI 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?