Replaced the nb of batches arguments with nb of samples.