Tensorboard logging should use `num_grad_updates` not `batch_idx`
When accumulate_grad_batches > 1, the x-axis in tensorboard should be number of gradient updates, not number of batches that have been processed.