keras-unet-collection
keras-unet-collection copied to clipboard
How to add dropout and reguralizer to the unet_2d model
I am trying to use the unet_2d model from collection . However the curve shows that data is overfitting. I want to add regularizer and droput to the layers . How can I incorporate this ?