Skip to content

issue about model.py/dropout = tf.layers.dropout(pool, rate=drop_rate) #16

Description

@YanShuang17

Hello, when i train the model with my own dataset, i found the statement:
dropout = tf.layers.dropout(pool, rate=drop_rate)
in model.py may be have problem.
The operation: tf.layers.dropout() in TensorFlow's document has an argument of training, and training=False by default, in your code, you didn't set the training argument, so, although you set the rate value, the model won't conduct the dropout operation???

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions