You can change the prefetch_factor, which is defined as:

prefetch_factor (int, optional , keyword-only arg) – Number of samples loaded in advance by each worker. 2 means there will be a total of 2 * num_workers samples prefetched across all workers. (default: 2)