fit_image_data_generator(object, x, augment = FALSE, rounds = 1, seed = NULL)
array, the data to fit on (should have rank 4). In case of grayscale data, the channels axis should have value 1, and in case of RGB data, it should have value 3.
Whether to fit on randomly augmented samples
augment, how many augmentation passes to do over the data