```plot_data_generator(num_samples=3, fig_size=(10, 10), train_config=config, img_ch=3)``` 