tomodrgn.commands.train\_vae
============================

.. automodule:: tomodrgn.commands.train_vae

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree:
      :nosignatures:
   
      add_args
      decode_batch
      encode_batch
      encoder_inference
      loss_function
      main
      preprocess_batch
      save_checkpoint
      train_batch