urbanlc.model.deep_learning.MSSDeepLearning.set_device

MSSDeepLearning.set_device(device: Union[None, str]) → None

Set the device for training or inference. Defaults to “cuda” if available.

Parameters:

device (Union[None, str]) – Device to use for training (e.g., “cuda” or “cpu”).