def __init__(self, params, mode, name): GraphModule.__init__(self, name) Configurable.__init__(self, params, mode) self.params["rnn_cell"] = _toggle_dropout(self.params["rnn_cell"], mode) self.cell = training_utils.get_rnn_cell(**self.params["rnn_cell"]) # Not initialized yet self.initial_state = None self.helper = None
def __init__(self, params, mode, name): GraphModule.__init__(self, name) Configurable.__init__(self, params, mode) self.params["rnn_cell"] = _toggle_dropout(self.params["rnn_cell"], mode) self.cell = training_utils.get_rnn_cell(**self.params["rnn_cell"]) # Not initialized yet self.initial_state = None self.helper = None
def __init__(self, params, mode, name): GraphModule.__init__(self, name) Configurable.__init__(self, params, mode) self.params["rnn_cell"] = _toggle_dropout( self.params["rnn_cell"], ["dropout_input_keep_prob", "dropout_output_keep_prob"], mode) self.params["dropout_decoder"] = _toggle_dropout( self.params["dropout_decoder"], ["dropout_deep_output_layer"], mode) with tf.variable_scope("rec1"): self.cell_rec1 = training_utils.get_rnn_cell( **self.params["rnn_cell"]) with tf.variable_scope("rec2"): self.cell_rec2 = training_utils.get_rnn_cell( **self.params["rnn_cell"]) # Not initialized yet self.initial_state = None self.helper = None