add check size of both datasets in training with early stopping