ms
Iteration 800/1000, avg batch time 362.96 ms
Iteration 900/1000, avg batch time 362.96 ms
Iteration 1000/1000, avg batch time 362.99 ms
Input shape: torch.Size([128, 3, 300, 300])
Output location prediction size: torch.Size([128, 4, 8732])
Output label prediction size: torch.Size([128, 81, 8732])
Average batch time: 362.99 ms
</pre>
         </div>
        </div>
       </div>
       <hr class="docutils"/>
       <p>
        ## 3. Creating TorchScript modules
       </p>
       <p>
        To compile wit