tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning
Apache License 2.0
3.18k stars 439 forks source link

Droupout probabilities are trainable #100

Closed olegarch closed 8 years ago

olegarch commented 8 years ago

Shouldn't it be like this?

prob = tf.get_variable("prob", [], initializer=tf.constant_initializer(prob), trainable=False)

ilblackdragon commented 8 years ago

That's a very good point. I'll fix it now.