Fixed iterator to calculate max labels before taking subset, training using validation set, fixed number of batches