Browse Source

translate chinese comment to english

master
智丞 3 years ago
parent
commit
1e361a4ad1
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      modelscope/models/nlp/space/model/generator.py

+ 2
- 1
modelscope/models/nlp/space/model/generator.py View File

@@ -183,7 +183,8 @@ class BeamSearch(Generator):

scores_after_end = np.full(self.vocab_size, -1e10, dtype='float32')
scores_after_end[
self.pad_id] = 0 # we want <pad> is generated after <eos>,so maximum log(p(<pad>)) is (0)
self.
pad_id] = 0 # we want <pad> is generated after <eos>,so maximum log(p(<pad>)) is (0)
scores_after_end = torch.from_numpy(scores_after_end)

if self.use_gpu:


Loading…
Cancel
Save