Set the number of test files in a batch to match the sgd batch size