lass="n">IN_CH</span> <span class="o">=</span> <span class="mi">3</span>
<span class="n">OUT_CH</span> <span class="o">=</span> <span class="mi">128</span>
<span class="n">BATCH_SIZE</span> <span class="o">=</span> <span class="mi">1</span>

<span class="c1"># set seed</span>
<span class="n">torch</span><span class="o">.</span><span class="n">manual_seed</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span>

<span class="n">linear</span> <span class="o">=</span> <span class="n">qnn