fixed batch size to be changed to 1 for actually running the model rather then 20 when training