We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 373776b commit 35de885Copy full SHA for 35de885
beginner_source/t5_tutorial.py
@@ -406,7 +406,7 @@ def process_labels(labels, x):
406
beam_size = 4
407
408
model_input = transform(input_text)
409
-model_output = sequence_generator.generate(model_input, eos_idx=eos_idx, beam_size=beam_size)
+model_output = sequence_generator.generate(model_input, eos_idx=eos_idx, num_beams=beam_size)
410
output_text = transform.decode(model_output.tolist())
411
412
for i in range(multi_batch_size):
0 commit comments