Getting Started
Documentation
Development
deepnog.utils
Author: Roman Feldbauer
Date: 2020-02-19
Description:
Various utility functions
deepnog.utils.
set_device
Set device (CPU/GPU) depending on user choice and availability.
device (str) – Device set by user as an argument to DeepNOG call.
device – Object containing the device type to be used for prediction calculations.
torch.device