discopat.utils

Functions

get_device([allow_mps])

discopat.utils.get_device(allow_mps=True)[source]
Parameters:

allow_mps (bool)