dfreproject.utils
=================

.. py:module:: dfreproject.utils


Functions
---------

.. autoapisummary::

   dfreproject.utils.get_device


Module Contents
---------------

.. py:function:: get_device()

   Utility function to get the device

   :returns: Available torch device (either gpu or cpu)
   :rtype: torch.device