修改output max的错误

translate_file2.py

@@ -72,6 +72,8 @@ def get_result(src, model, SRC, TRG, opt):
     if opt.tone_filter == True:
         output_min, _ = output.min(dim=-1)
         output_min = output_min.min()
+        output_max, _ = output.max(dim=-1)
+        output_max = output_max.max()
         vals, indices = output.topk(k=10, dim=-1, largest=True, sorted=True)
         result = []
         for x, index in enumerate(src[0][:]):
@@ -94,6 +96,7 @@ def get_result(src, model, SRC, TRG, opt):
                             output[0, x, indices[0][x][i]] = output_min
                     if not flag:
                         result.append(TRG.vocab.itos[indices[0][x][0]])
+                        output[0, x, indices[0][x][0]] = output_max
                 # 查看之前的声母
                 else:
                     result = result[:-1]
@@ -112,9 +115,10 @@ def get_result(src, model, SRC, TRG, opt):
                         # if flag:
                         #     break
                         if not flag:
-                            output[0, x, indices[0][x][i]] = output_min
+                            output[0, x-1, indices[0][x-1][i]] = output_min
                     if not flag:
                         result.append(TRG.vocab.itos[indices[0][x-1][0]])
+                        output[0, x, indices[0][x-1][0]] = output_max
                         result.append(TRG.vocab.itos[indices[0][x][0]])
             else:
                 result.append(TRG.vocab.itos[indices[0][x][0]])