Skip to content

Commit

Permalink
add eos layer as an extra parent.
Browse files Browse the repository at this point in the history
  • Loading branch information
lcy-seso committed Jun 7, 2017
1 parent 1e21259 commit 35332a2
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions python/paddle/v2/layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,6 +132,13 @@ def add_additional_parents():
add_parent(mem.layer_name, mem.boot_layer_name)
add_parent(mem.link_name, mem.layer_name)

if sub_model.HasField('generator'):
# according to the implementation of text generation
# in recurrent layer group, the generated word must be
# the first out link
add_parent(sub_model.out_links[0].layer_name,
sub_model.generator.eos_layer_name)

def dfs_travel(layer_name):
if layer_name in layer_names:
return
Expand Down Expand Up @@ -175,8 +182,6 @@ def __get_used_submodels__(layer_names):
for submodel in cp.g_config.model_config.sub_models:
if submodel.name in layer_names:
submodel_names.add(submodel.name)
if submodel.is_recurrent_layer_group:
layer_names |= set(submodel.layer_names)
return submodel_names


Expand Down

0 comments on commit 35332a2

Please sign in to comment.