From ebc3e5f35fb4974f3a5d442a2a0130d766a1d6e2 Mon Sep 17 00:00:00 2001 From: wanghy9512 <767105779@qq.com> Date: Sun, 22 Apr 2018 00:51:27 +0800 Subject: [PATCH] Update core.py --- tensorlayer/layers/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)