Iteration Time in PyTorch Scenarios