ggt.utils.model_utils
=====================

.. py:module:: ggt.utils.model_utils


Functions
---------

.. autoapisummary::

   ggt.utils.model_utils.get_output_shape
   ggt.utils.model_utils.enable_dropout
   ggt.utils.model_utils.specify_dropout_rate


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

.. py:function:: get_output_shape(model, image_dim)

   Get output shape of a PyTorch model or layer


.. py:function:: enable_dropout(model)

   Enable random dropout during inference. From StackOverflow #63397197


.. py:function:: specify_dropout_rate(model, rate)

   Specify the dropout rate of all layers