span class="p">,</span> <span class="s2">&quot;beta_2&quot;</span><span class="p">:</span> <span class="mf">0.97</span><span class="p">},</span>
    <span class="s2">&quot;name&quot;</span><span class="p">:</span> <span class="s2">&quot;adamax&quot;</span>
  <span class="p">}</span>
<span class="p">}</span>

<span class="n">vae</span> <span class="o">=</span> <span class="n">VAE</span><span class="p">(</span><span class="n">dataset</span><span class="p">,</span> <span class="n">dataset</span><span class="p"