How to add more regularization methods?

MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++

Other

2.26k stars 538 forks source link

How to add more regularization methods? #32

Open qiang-zhang-neu opened 4 years ago

qiang-zhang-neu commented 4 years ago

This project really help me a lot. However, I meet a problem when training our own data that the model is overfitted. Thus, I want to add more regularization methods, like L1/L2 or dropout. How can I add those features?