
Iteration 300/1000, avg batch time 363.09 ms
Iteration 400/1000, avg batch time 363.05 ms
Iteration 500/1000, avg batch time 363.08 ms
Iteration 600/1000, avg batch time 363.07 ms
Iteration 700/1000, avg batch time 363.09 ms
Iteration 800/1000, avg batch time 363.06 ms
Iteration 900/1000, avg batch time 363.08 ms
Iteration 1000/1000, avg batch time 363.08 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,