diff --git a/tensorlayer/layers/core.py b/tensorlayer/layers/core.py index b502fcc4c..80ac83a22 100644 --- a/tensorlayer/layers/core.py +++ b/tensorlayer/layers/core.py @@ -1280,7 +1280,7 @@ def __init__( if is_fix: self.outputs = tf.nn.dropout(self.inputs, keep, seed=seed, name=name) else: - LayersConfig.set_keep[name] = tf.placeholder(tf.float32) + LayersConfig.set_keep[name] = tf.placeholder(LayersConfig.tf_dtype) self.outputs = tf.nn.dropout(self.inputs, LayersConfig.set_keep[name], seed=seed, name=name) # 1.2 # self.all_layers = list(layer.all_layers)