Use DataSetIterator instead of Dataset
Implement the usage of DataSetIterator to allow the change of batchsize, epoch number,... Currently, all dataset is a single batch. It is not really good for optimization
Implement the usage of DataSetIterator to allow the change of batchsize, epoch number,... Currently, all dataset is a single batch. It is not really good for optimization