about requirements problems

[Moyao6]

Thanks for sharing your code.
When I tried to use the given requirements, there were some problems occurs.
I managed to train with these version:
numpy: 1.18.5
tensorflow-gpu: 2.3.0
keras: 2.4.3
segmentation-models: 1.0.1
CUDA Version 10.1.243

you can use this images to get all these environments:
docker pull smitharauco/tensorflow2.3gpu

After using these environments, you should insert one line of code in model.py
from keras.utils import generic_utils
before:
import segmentation_models as sm