batch size definition in exp setup

batch size should be defined in the experiment setup instead of inside the model class