deepnog.utils

Author: Roman Feldbauer

Date: 2020-02-19

Description:

Various utility functions

deepnog.utils.set_device(device)[source]

Set device (CPU/GPU) depending on user choice and availability.

Parameters

device (str) – Device set by user as an argument to DeepNOG call.

Returns

device – Object containing the device type to be used for prediction calculations.

Return type

torch.device