Data type and device transfers

metatrain.utils.transfer.systems_and_targets_to_device(systems: List[System], targets: Dict[str, TensorMap], device: device)

Transfers the systems and targets to the specified device.

Parameters:

• systems (List[System]) – List of systems.

• targets (Dict[str, TensorMap]) – Dictionary of targets.

• device (device) – Device to transfer to.

metatrain.utils.transfer.systems_and_targets_to_dtype(systems: List[System], targets: Dict[str, TensorMap], dtype: dtype)

Changes the systems and targets to the specified floating point data type.

Parameters:

• systems (List[System]) – List of systems.

• targets (Dict[str, TensorMap]) – Dictionary of targets.

• dtype (dtype) – Desired floating point data type.