dfreproject.utils

Functions

get_device()

Utility function to get the device

Module Contents

dfreproject.utils.get_device()

Utility function to get the device

Returns:

Available torch device (either gpu or cpu)

Return type:

torch.device