Log information about training datasets

We need to know how many elements were used in the datasets. Please add a log before training starts specifying the size of both the training and the validation datasets.