"nam">self</span><span class="op">.</span><span class="nam">train_loss_buffer</span><span class="op">[</span><span class="op">:</span><span class="op">,</span><span class="nam">self</span><span class="op">.</span><span class="nam">epoch</span><span class="op">-</span><span class="num">2</span><span class="op">]</span><span class="op">)</span><span class="op">.</span><span class="nam">to</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">device</span><span 