from local

This commit is contained in:
dohe0342 2023-02-25 16:27:29 +09:00
parent e67de22873
commit 8468e23874
4 changed files with 1 additions and 2 deletions

View File

@ -425,7 +425,6 @@ def decode_one_batch(
eos_id=graph_compiler.eos_id,
return_output=True,
)
hyps = greedy_search(pred, memory_key_padding_mask)
# hyps is a list of str, e.g., ['xxx yyy zzz', ...]

View File

@ -882,7 +882,7 @@ class TransformerDecoder(nn.Module):
if i in self.aux_layers:
outputs.append(output)
output = self.combiner(outputs)
#output = self.combiner(outputs)
return output