ggt.visualization.spatial_transform
===================================

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


Attributes
----------

.. autoapisummary::

   ggt.visualization.spatial_transform.log_fmt


Functions
---------

.. autoapisummary::

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


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

.. 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'