deploy: 0ecf72f

liopeer · Nov 10, 2023 · 2bd8089 · 2bd8089
1 parent 5de8036
commit 2bd8089
Show file tree

Hide file tree

Showing 48 changed files with 1,337 additions and 102 deletions.
diff --git a/_autosummary/diffusion_models.losses.kl_divergence.gaussian_kl.html b/_autosummary/diffusion_models.losses.kl_divergence.gaussian_kl.html
@@ -111,10 +111,10 @@ <h1>diffusion_models.losses.kl_divergence.gaussian_kl<a class="headerlink" href=
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>p_mean</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – mean value of first distribution</p></li>
-<li><p><strong>p_var</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – variance value of first distribution</p></li>
-<li><p><strong>q_mean</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – mean value of second distribution</p></li>
-<li><p><strong>q_var</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – variance value of second distribution</p></li>
+<li><p><strong>p_mean</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – mean value of first distribution</p></li>
+<li><p><strong>p_var</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – variance value of first distribution</p></li>
+<li><p><strong>q_mean</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – mean value of second distribution</p></li>
+<li><p><strong>q_var</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – variance value of second distribution</p></li>
 </ul>
 </dd>
 <dt class="field-even">Returns<span class="colon">:</span></dt>

diff --git a/_autosummary/diffusion_models.losses.kl_divergence.log_gaussian_kl.html b/_autosummary/diffusion_models.losses.kl_divergence.log_gaussian_kl.html
@@ -111,10 +111,10 @@ <h1>diffusion_models.losses.kl_divergence.log_gaussian_kl<a class="headerlink" h
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>p_mean</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – mean value of first distribution</p></li>
-<li><p><strong>p_logvar</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – log of variance value of first distribution</p></li>
-<li><p><strong>q_mean</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – mean value of second distribution</p></li>
-<li><p><strong>q_logvar</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code>) – log of variance value of second distribution</p></li>
+<li><p><strong>p_mean</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – mean value of first distribution</p></li>
+<li><p><strong>p_logvar</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – log of variance value of first distribution</p></li>
+<li><p><strong>q_mean</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – mean value of second distribution</p></li>
+<li><p><strong>q_logvar</strong> (<span class="sphinx_autodoc_typehints-type"><code class="xref py py-class docutils literal notranslate"><span class="pre">Float[Tensor,</span> <span class="pre">'1']</span></code></span>) – log of variance value of second distribution</p></li>
 </ul>
 </dd>
 <dt class="field-even">Returns<span class="colon">:</span></dt>

diff --git a/_autosummary/diffusion_models.models.html b/_autosummary/diffusion_models.models.html
@@ -49,6 +49,7 @@
 <li class="toctree-l1 current"><a class="reference internal" href="../api.html">API Reference</a><ul class="current">
 <li class="toctree-l2 current"><a class="current reference internal" href="#">diffusion_models.models</a><ul>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.mnist_enc.html">diffusion_models.models.mnist_enc</a></li>
+<li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.multicoil.html">diffusion_models.models.multicoil</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.positional_encoding.html">diffusion_models.models.positional_encoding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.unet.html">diffusion_models.models.unet</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.vae.html">diffusion_models.models.vae</a></li>
@@ -105,13 +106,16 @@
 <tr class="row-odd"><td><p><a class="reference internal" href="diffusion_models.models.mnist_enc.html#module-diffusion_models.models.mnist_enc" title="diffusion_models.models.mnist_enc"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.mnist_enc</span></code></a></p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="diffusion_models.models.positional_encoding.html#module-diffusion_models.models.positional_encoding" title="diffusion_models.models.positional_encoding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.positional_encoding</span></code></a></p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="diffusion_models.models.multicoil.html#module-diffusion_models.models.multicoil" title="diffusion_models.models.multicoil"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.multicoil</span></code></a></p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="diffusion_models.models.unet.html#module-diffusion_models.models.unet" title="diffusion_models.models.unet"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.unet</span></code></a></p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="diffusion_models.models.positional_encoding.html#module-diffusion_models.models.positional_encoding" title="diffusion_models.models.positional_encoding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.positional_encoding</span></code></a></p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="diffusion_models.models.vae.html#module-diffusion_models.models.vae" title="diffusion_models.models.vae"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.vae</span></code></a></p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="diffusion_models.models.unet.html#module-diffusion_models.models.unet" title="diffusion_models.models.unet"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.unet</span></code></a></p></td>
+<td><p></p></td>
+</tr>
+<tr class="row-odd"><td><p><a class="reference internal" href="diffusion_models.models.vae.html#module-diffusion_models.models.vae" title="diffusion_models.models.vae"><code class="xref py py-obj docutils literal notranslate"><span class="pre">diffusion_models.models.vae</span></code></a></p></td>
 <td><p></p></td>
 </tr>
 </tbody>

diff --git a/_autosummary/diffusion_models.models.mnist_enc.MNISTEncoder.html b/_autosummary/diffusion_models.models.mnist_enc.MNISTEncoder.html
@@ -19,7 +19,7 @@
     <script src="../_static/js/theme.js"></script>
     <link rel="index" title="Index" href="../genindex.html" />
     <link rel="search" title="Search" href="../search.html" />
-    <link rel="next" title="diffusion_models.models.positional_encoding" href="diffusion_models.models.positional_encoding.html" />
+    <link rel="next" title="diffusion_models.models.multicoil" href="diffusion_models.models.multicoil.html" />
     <link rel="prev" title="diffusion_models.models.mnist_enc" href="diffusion_models.models.mnist_enc.html" /> 
 </head>
 
@@ -52,6 +52,7 @@
 <li class="toctree-l4 current"><a class="current reference internal" href="#">diffusion_models.models.mnist_enc.MNISTEncoder</a></li>
 </ul>
 </li>
+<li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.multicoil.html">diffusion_models.models.multicoil</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.positional_encoding.html">diffusion_models.models.positional_encoding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.unet.html">diffusion_models.models.unet</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.vae.html">diffusion_models.models.vae</a></li>
@@ -293,7 +294,7 @@ <h1>diffusion_models.models.mnist_enc.MNISTEncoder<a class="headerlink" href="#d
           </div>
           <footer><div class="rst-footer-buttons" role="navigation" aria-label="Footer">
         <a href="diffusion_models.models.mnist_enc.html" class="btn btn-neutral float-left" title="diffusion_models.models.mnist_enc" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
-        <a href="diffusion_models.models.positional_encoding.html" class="btn btn-neutral float-right" title="diffusion_models.models.positional_encoding" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
+        <a href="diffusion_models.models.multicoil.html" class="btn btn-neutral float-right" title="diffusion_models.models.multicoil" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
     </div>
 
   <hr/>

diff --git a/_autosummary/diffusion_models.models.mnist_enc.html b/_autosummary/diffusion_models.models.mnist_enc.html
@@ -52,6 +52,7 @@
 <li class="toctree-l4"><a class="reference internal" href="diffusion_models.models.mnist_enc.MNISTEncoder.html">diffusion_models.models.mnist_enc.MNISTEncoder</a></li>
 </ul>
 </li>
+<li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.multicoil.html">diffusion_models.models.multicoil</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.positional_encoding.html">diffusion_models.models.positional_encoding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.unet.html">diffusion_models.models.unet</a></li>
 <li class="toctree-l3"><a class="reference internal" href="diffusion_models.models.vae.html">diffusion_models.models.vae</a></li>