[layers] bidynamicrnn, out_dropout_keep

Author: zsdonghao

tensorlayer/layers.py

@@ -4739,6 +4739,13 @@ def __init__(
                 # cell_instance_fn=lambda: MultiRNNCell_fn([cell_instance_fn2() for _ in range(n_layer)])
                 self.fw_cell = MultiRNNCell_fn([cell_creator() for _ in range(n_layer)])
                 self.bw_cell = MultiRNNCell_fn([cell_creator() for _ in range(n_layer)])
+
+            if dropout:
+                self.fw_cell = DropoutWrapper_fn(self.fw_cell,
+                          input_keep_prob=1.0, output_keep_prob=out_keep_prob)
+                self.bw_cell = DropoutWrapper_fn(self.bw_cell,
+                          input_keep_prob=1.0, output_keep_prob=out_keep_prob)
+
             # self.fw_cell=cell_instance_fn()
             # self.bw_cell=cell_instance_fn()
             # Initial state of RNN

tensorlayer/ops.py

@@ -43,24 +43,23 @@ def exit_tf(sess=None, port=6006):
         # TODO
     else:
         print(text2 + _platform)
-    
+
 def open_tb(logdir='/tmp/tensorflow', port=6006):
-    """Open tensorboard
-    
+    """Open Tensorboard.
+
     Parameters
     ----------
     logdir : a string
         Directory where your tensorboard logs are saved
     port : an integer
         TensorBoard port you want to open, 6006 is tensorboard default
     """
-
     text = "[tl] Open tensorboard, go to localhost:" + str(port) + " to access"
     text2 = " not yet supported by this function (tl.ops.open_tb)"
-    
+
     if not tl.files.exists_or_mkdir(logdir, verbose=False):
         print("[tl] Log reportory was created at %s" % logdir)
-    
+
     if _platform == "linux" or _platform == "linux2":
         print('linux %s' % text2)
         # TODO