Skip to content

Commit f188e89

Browse files
author
gongel
committed
refactor: default to beam search v1
1 parent 6414565 commit f188e89

File tree

1 file changed

+1
-3
lines changed
  • examples/machine_translation/transformer

1 file changed

+1
-3
lines changed

examples/machine_translation/transformer/predict.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,7 @@ def do_predict(args):
8484
eos_id=args.eos_idx,
8585
beam_size=args.beam_size,
8686
max_out_len=args.max_out_len,
87-
use_ft=not args.without_ft,
88-
beam_search_version='v2',
89-
alpha=0.6)
87+
use_ft=not args.without_ft)
9088

9189
# Load the trained model
9290
assert args.init_from_params, (

0 commit comments

Comments
 (0)