katex fixes

Author: vpj

docs/diffusion/ddpm/evaluate.html

@@ -992,7 +992,7 @@ <h4>Sample from <span class="katex"><span aria-hidden="true" class="katex-html">
 
         </div>
         <div class='code'>
-            <div class="highlight"><pre><span class="lineno">253</span>        <span class="n">eps_coef</span> <span class="o">=</span> <span class="p">(</span><span class="mi">1</span> <span class="o">-</span> <span class="n">alpha</span><span class="p">)</span> <span class="o">/</span> <span class="p">(</span><span class="mi">1</span> <span class="o">-</span> <span class="n">alpha_bar</span><span class="p">)</span> <span class="o">**</span> <span class="o">.</span><span class="mi">5</span></pre></div>
+            <div class="highlight"><pre><span class="lineno">253</span>        <span class="n">eps_coef</span> <span class="o">=</span> <span class="p">(</span><span class="mi">1</span> <span class="o">-</span> <span class="n">alpha</span><span class="p">)</span> <span class="o">/</span> <span class="p">(</span><span class="mi">1</span> <span class="o">-</span> <span class="n">alpha_bar</span><span class="p">)</span> <span class="o">**</span> <span class="mf">.5</span></pre></div>
         </div>
     </div>
     <div class='section' id='section-64'>
@@ -1062,7 +1062,7 @@ <h4>Sample from <span class="katex"><span aria-hidden="true" class="katex-html">
 
         </div>
         <div class='code'>
-            <div class="highlight"><pre><span class="lineno">263</span>        <span class="k">return</span> <span class="n">mean</span> <span class="o">+</span> <span class="p">(</span><span class="n">var</span> <span class="o">**</span> <span class="o">.</span><span class="mi">5</span><span class="p">)</span> <span class="o">*</span> <span class="n">eps</span></pre></div>
+            <div class="highlight"><pre><span class="lineno">263</span>        <span class="k">return</span> <span class="n">mean</span> <span class="o">+</span> <span class="p">(</span><span class="n">var</span> <span class="o">**</span> <span class="mf">.5</span><span class="p">)</span> <span class="o">*</span> <span class="n">eps</span></pre></div>
         </div>
     </div>
     <div class='section' id='section-68'>

docs/diffusion/ddpm/index.html

@@ -142,12 +142,12 @@ <h3>Installation</h3>
 <pre class="highlight lang-bash"><code><span></span>pip install labml-nn</code></pre>
 <h3>Citing LabML</h3>
 <p>If you use this for academic research, please cite it using the following BibTeX entry.</p>
-<pre class="highlight lang-bibtex"><code><span></span><span class="nc">@misc</span><span class="p">{</span><span class="nl">labml</span><span class="p">,</span>
- <span class="na">author</span> <span class="p">=</span> <span class="s">{Varuna Jayasiri, Nipun Wijerathne}</span><span class="p">,</span>
- <span class="na">title</span> <span class="p">=</span> <span class="s">{labml.ai Annotated Paper Implementations}</span><span class="p">,</span>
- <span class="na">year</span> <span class="p">=</span> <span class="s">{2020}</span><span class="p">,</span>
- <span class="na">url</span> <span class="p">=</span> <span class="s">{https://nn.labml.ai/}</span><span class="p">,</span>
-<span class="p">}</span></code></pre>
+<pre class="highlight lang-bibtex"><code><span></span><span class="nc">@misc</span><span class="p">{</span><span class="nl">labml</span><span class="p">,</span><span class="w"></span>
+<span class="w"> </span><span class="na">author</span><span class="w"> </span><span class="p">=</span><span class="w"> </span><span class="s">{Varuna Jayasiri, Nipun Wijerathne}</span><span class="p">,</span><span class="w"></span>
+<span class="w"> </span><span class="na">title</span><span class="w"> </span><span class="p">=</span><span class="w"> </span><span class="s">{labml.ai Annotated Paper Implementations}</span><span class="p">,</span><span class="w"></span>
+<span class="w"> </span><span class="na">year</span><span class="w"> </span><span class="p">=</span><span class="w"> </span><span class="s">{2020}</span><span class="p">,</span><span class="w"></span>
+<span class="w"> </span><span class="na">url</span><span class="w"> </span><span class="p">=</span><span class="w"> </span><span class="s">{https://nn.labml.ai/}</span><span class="p">,</span><span class="w"></span>
+<span class="p">}</span><span class="w"></span></code></pre>
 
         </div>
         <div class='code'>

docs/index.html

@@ -306,7 +306,7 @@ <h2>Synthetic Experiment</h2>
 
         </div>
         <div class='code'>
-            <div class="highlight"><pre><span class="lineno">153</span>    <span class="n">x</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Parameter</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">([</span><span class="o">.</span><span class="mi">0</span><span class="p">]))</span></pre></div>
+            <div class="highlight"><pre><span class="lineno">153</span>    <span class="n">x</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Parameter</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">([</span><span class="mf">.0</span><span class="p">]))</span></pre></div>
         </div>
     </div>
     <div class='section' id='section-15'>

docs/optimizers/amsgrad.html

@@ -997,7 +997,7 @@ <h3>Calculate total loss</h3>
 
         </div>
         <div class='code'>
-            <div class="highlight"><pre><span class="lineno">313</span>        <span class="n">approx_kl_divergence</span> <span class="o">=</span> <span class="o">.</span><span class="mi">5</span> <span class="o">*</span> <span class="p">((</span><span class="n">samples</span><span class="p">[</span><span class="s1">&#39;log_pis&#39;</span><span class="p">]</span> <span class="o">-</span> <span class="n">log_pi</span><span class="p">)</span> <span class="o">**</span> <span class="mi">2</span><span class="p">)</span><span class="o">.</span><span class="n">mean</span><span class="p">()</span></pre></div>
+            <div class="highlight"><pre><span class="lineno">313</span>        <span class="n">approx_kl_divergence</span> <span class="o">=</span> <span class="mf">.5</span> <span class="o">*</span> <span class="p">((</span><span class="n">samples</span><span class="p">[</span><span class="s1">&#39;log_pis&#39;</span><span class="p">]</span> <span class="o">-</span> <span class="n">log_pi</span><span class="p">)</span> <span class="o">**</span> <span class="mi">2</span><span class="p">)</span><span class="o">.</span><span class="n">mean</span><span class="p">()</span></pre></div>
         </div>
     </div>
     <div class='section' id='section-71'>