an class="c1"># Otherwise, calculate self-supervised loss</span>
            <span class="n">self_sup_output</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">self_supervised_loss</span><span class="p">(</span>
                <span class="n">batch</span><span class="p">[</span><span class="s1">&#39;rgb_original&#39;</span><span class="p">],</span> <span class="n">batch</span><span class="p">[</span><span class="s1">&#39;rgb_context_original&#39;</span><span