tiny\_diff.models.vae.model.ConvVAE
===================================

.. currentmodule:: tiny_diff.models.vae.model

.. autoclass:: ConvVAE
   :members:                                    <-- add at least this line
   :show-inheritance:                           <-- plus I want to show inheritance...
   :no-inherited-members:                          <-- ...and inherited members too

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~ConvVAE.__init__
      ~ConvVAE.add_module
      ~ConvVAE.apply
      ~ConvVAE.bfloat16
      ~ConvVAE.buffers
      ~ConvVAE.children
      ~ConvVAE.compile
      ~ConvVAE.cpu
      ~ConvVAE.cuda
      ~ConvVAE.double
      ~ConvVAE.eval
      ~ConvVAE.extra_repr
      ~ConvVAE.float
      ~ConvVAE.forward
      ~ConvVAE.forward_decode
      ~ConvVAE.forward_encode
      ~ConvVAE.get_buffer
      ~ConvVAE.get_extra_state
      ~ConvVAE.get_parameter
      ~ConvVAE.get_submodule
      ~ConvVAE.half
      ~ConvVAE.ipu
      ~ConvVAE.latent_sample
      ~ConvVAE.load_state_dict
      ~ConvVAE.loss
      ~ConvVAE.loss_with_fwd
      ~ConvVAE.modules
      ~ConvVAE.named_buffers
      ~ConvVAE.named_children
      ~ConvVAE.named_modules
      ~ConvVAE.named_parameters
      ~ConvVAE.parameters
      ~ConvVAE.register_backward_hook
      ~ConvVAE.register_buffer
      ~ConvVAE.register_forward_hook
      ~ConvVAE.register_forward_pre_hook
      ~ConvVAE.register_full_backward_hook
      ~ConvVAE.register_full_backward_pre_hook
      ~ConvVAE.register_load_state_dict_post_hook
      ~ConvVAE.register_module
      ~ConvVAE.register_parameter
      ~ConvVAE.register_state_dict_pre_hook
      ~ConvVAE.reparameterize
      ~ConvVAE.requires_grad_
      ~ConvVAE.sample
      ~ConvVAE.save
      ~ConvVAE.set_extra_state
      ~ConvVAE.share_memory
      ~ConvVAE.state_dict
      ~ConvVAE.to
      ~ConvVAE.to_empty
      ~ConvVAE.train
      ~ConvVAE.type
      ~ConvVAE.xpu
      ~ConvVAE.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~ConvVAE.T_destination
      ~ConvVAE.call_super_init
      ~ConvVAE.dump_patches
      ~ConvVAE.sample_channels
      ~ConvVAE.training