tiny\_diff.models.unet.model.UNet
=================================

.. currentmodule:: tiny_diff.models.unet.model

.. autoclass:: UNet
   :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::
   
      ~UNet.__init__
      ~UNet.add_module
      ~UNet.apply
      ~UNet.bfloat16
      ~UNet.buffers
      ~UNet.children
      ~UNet.compile
      ~UNet.conv_kwargs
      ~UNet.cpu
      ~UNet.cuda
      ~UNet.double
      ~UNet.eval
      ~UNet.extra_repr
      ~UNet.float
      ~UNet.forward
      ~UNet.get_buffer
      ~UNet.get_extra_state
      ~UNet.get_parameter
      ~UNet.get_submodule
      ~UNet.half
      ~UNet.ipu
      ~UNet.load_state_dict
      ~UNet.modules
      ~UNet.named_buffers
      ~UNet.named_children
      ~UNet.named_modules
      ~UNet.named_parameters
      ~UNet.parameters
      ~UNet.register_backward_hook
      ~UNet.register_buffer
      ~UNet.register_forward_hook
      ~UNet.register_forward_pre_hook
      ~UNet.register_full_backward_hook
      ~UNet.register_full_backward_pre_hook
      ~UNet.register_load_state_dict_post_hook
      ~UNet.register_module
      ~UNet.register_parameter
      ~UNet.register_state_dict_pre_hook
      ~UNet.requires_grad_
      ~UNet.res_kwargs
      ~UNet.save
      ~UNet.set_extra_state
      ~UNet.share_memory
      ~UNet.state_dict
      ~UNet.to
      ~UNet.to_empty
      ~UNet.train
      ~UNet.type
      ~UNet.up_res_kwargs
      ~UNet.xpu
      ~UNet.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~UNet.T_destination
      ~UNet.call_super_init
      ~UNet.channels
      ~UNet.conv_channels
      ~UNet.down_channels
      ~UNet.dump_patches
      ~UNet.n_conv_layers
      ~UNet.r_channels
      ~UNet.scale_factor
      ~UNet.stride
      ~UNet.up_channels
      ~UNet.training