      <span class="nb">print</span><span class="p">(</span><span class="s1">&#39;Save model.&#39;</span><span class="p">)</span>
                <span class="n">torch</span><span class="o">.</span><span class="n">save</span><span class="p">(</span><span class="n">train_net</span><span class="o">.</span><span class="n">state_dict</span><span class="p">(),</span> <span class="s1">&#39;./trained_models/&#39;</span><span class="o">+</span> <span class="n">data</span> <span class="o">+</span> <span class="s2">&q