Some tweaks

- log the time along with the batch size
- change the batch size to 500