gzuidhof / luna16

LUNA16 Lung Nodule Analysis - NWI-IMC037 Final Project
BSD 2-Clause "Simplified" License
185 stars 73 forks source link

SpatialDropout in Unet? where? #8

Closed rayanelleuch closed 7 years ago

rayanelleuch commented 7 years ago

On the paper released it says that SpatialDropout was used in the Unet model. But in the current code I just see the use of normal DropoutLayer. I saw that there is the implementation of SpatialDropout in custom_layers.py but I don't see it used. Is the code exactly the same used for Luna challenge?

gzuidhof commented 7 years ago

Hi Rayan,

You are correct, it seems ordinary dropout was used instead of spatial dropout due to a mistake in the code :confused:.

rayanelleuch commented 7 years ago

Ok thanks! I try to reproduce it on tensorflow but still kind of overfitting a lot from my point of view with spatial dropout. I will try with the normal dropout.