the dropout probability should be different between train and inference by zhengjxu · Pull Request #51 · golbin/TensorFlow-Tutorials · GitHub
Skip to content

the dropout probability should be different between train and inference#51

Merged
golbin merged 1 commit into
golbin:masterfrom
zhengjxu:master
Jan 4, 2023
Merged

the dropout probability should be different between train and inference#51
golbin merged 1 commit into
golbin:masterfrom
zhengjxu:master

Conversation

@zhengjxu

Copy link
Copy Markdown
Contributor

During the training, the keep_prob = 0.5, during inference, the keep_prob should be 1. I update the code by adding the keep_prob as tf.placeholder.

@golbin golbin merged commit c8ac1e4 into golbin:master Jan 4, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants