I use TFF and My dataset has a binary_mode class, this is how I declared my inputs :
genv0 = img_genv.flow_from_directory(pathv0,(224, 224),\'rgb\', batch_si