Skip to content

Commit 70bc62f

Browse files
wanghaoyang1995zsdonghao
authored andcommitted
dropout keep prob use config dtype (#519)
1 parent 025feef commit 70bc62f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorlayer/layers/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1280,7 +1280,7 @@ def __init__(
12801280
if is_fix:
12811281
self.outputs = tf.nn.dropout(self.inputs, keep, seed=seed, name=name)
12821282
else:
1283-
LayersConfig.set_keep[name] = tf.placeholder(tf.float32)
1283+
LayersConfig.set_keep[name] = tf.placeholder(LayersConfig.tf_dtype)
12841284
self.outputs = tf.nn.dropout(self.inputs, LayersConfig.set_keep[name], seed=seed, name=name) # 1.2
12851285

12861286
# self.all_layers = list(layer.all_layers)

0 commit comments

Comments
 (0)