Bug fix

- batch size was set to 50 when reading out data