="kn">import</span> <span class="nn">torch</span>
<span class="lineno">21</span><span class="kn">import</span> <span class="nn">torch.utils.data</span>
<span class="lineno">22</span><span class="kn">import</span> <span class="nn">torchvision.transforms.functional</span>
<span class="lineno">23</span><span class="kn">from</span> <span class="nn">torch</span> <span class="kn">import</span> <span class="n">nn</span>
<span class="lineno">24</span>
<span class="lineno">25</span><span class="kn">from</span> <span