Problem of 'cuDNN launch failure: input shape error'

Question

I get an error when I try to run a fit function for a model in an image segmentation project:

model.fit(data_generator_train, 
          steps_per_epoch=data_gene