Is it possible make that fit_generator?
I\'m creating a U-net network and I want to use as input a picture high 500, weight 500, and 5 channels, and on the output h