COSIE.COSIE_framework.COSIE_model.to_device

COSIE_model.to_device(device)[source]

Move all model parameters and submodules to the specified device.

Parameters

devicestr: Target device identifier (e.g., ‘cuda:0’, ‘cpu’).

Returns

None