SimpleUnet
==========

.. py:module:: SimpleUnet




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

.. py:class:: Block(in_ch, out_ch, time_emb_dim, up=False)



   .. py:attribute:: time_mlp


   .. py:attribute:: conv2


   .. py:attribute:: bnorm1


   .. py:attribute:: bnorm2


   .. py:attribute:: relu


   .. py:method:: forward(x, t)


.. py:class:: SinusoidalPositionEmbeddings(dim)



   .. py:attribute:: dim


   .. py:method:: forward(time)


.. py:class:: SimpleUnet(diffuser, image_channels)



   A simplified variant of the Unet architecture.


   .. py:attribute:: image_channels


   .. py:attribute:: down_channels
      :value: (64, 128, 256, 512, 1024)



   .. py:attribute:: up_channels
      :value: (1024, 512, 256, 128, 64)



   .. py:attribute:: out_dim


   .. py:attribute:: time_emb_dim
      :value: 32



   .. py:attribute:: time_mlp


   .. py:attribute:: conv0


   .. py:attribute:: downs


   .. py:attribute:: ups


   .. py:attribute:: output


   .. py:method:: forward(x, timestep)