Switched code logic to use batch_size of the length of the dataset.
Switched code logic to use batch_size of the length of the dataset.

MarcCote/smartlearner,SMART-Lab/smartpy,ASalvail/smartlearner,SMART-Lab/smartlearner