:py:mod:`ggt.visualization.spatial_transform`
=============================================

.. py:module:: ggt.visualization.spatial_transform


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


Functions
~~~~~~~~~

.. autoapisummary::

   ggt.visualization.spatial_transform.visualize_spatial_transform
   ggt.visualization.spatial_transform.main



Attributes
~~~~~~~~~~

.. autoapisummary::

   ggt.visualization.spatial_transform.log_fmt


.. py:function:: visualize_spatial_transform(model, loader, output_dir, device='cpu', nrow=6, return_grids=False, matplotlib_backend='agg')


.. py:function:: main(model_type, model_path, cutout_size, channels, n_out, data_dir, split_slug, split, batch_size, nrow, n_workers, normalize)

   Visualize the transformation performed by the spatial transformer
   module.


.. py:data:: log_fmt
   :value: '%(asctime)s - %(name)s - %(levelname)s - %(message)s'