fixed dropout/keep_prob confusion

pudae / tensorflow-densenet

Tensorflow-DenseNet with ImageNet Pretrained Models

Apache License 2.0

169 stars 59 forks source link

Open dmonn opened 5 years ago

dmonn commented 5 years ago

I saw this fix pop up in #11 and since it cost me a few training hours, decided to open a PR for it.

cag472 commented 5 years ago

-1 keep_prob is not defined. Should say 1.0-dropout_rate

dmonn commented 5 years ago

-1 keep_prob is not defined. Should say 1.0-dropout_rate

That's what it does. keep_prob = 1.0 - dropout_rate