00/1000, avg batch time 72.95 ms
Iteration 300/1000, avg batch time 73.00 ms
Iteration 400/1000, avg batch time 73.06 ms
Iteration 500/1000, avg batch time 73.10 ms
Iteration 600/1000, avg batch time 73.14 ms
Iteration 700/1000, avg batch time 73.17 ms
Iteration 800/1000, avg batch time 73.18 ms
Iteration 900/1000, avg batch time 73.19 ms
Iteration 1000/1000, avg batch time 73.21 ms
Input shape: torch.Size([128, 3, 300, 300])
Output location prediction size: torch.Size([128, 4, 8732])
Output label predictio