Unverified Commit f518228f authored by Piyush Thakur's avatar Piyush Thakur Committed by GitHub
Browse files

Fix incorrect `max_gen_toks` generation kwarg default in code2_text. (#1551)

* update gen_kwargs in code2-text-go.yaml

* update gen_kwargs in rest code2-text
parent 8051d954
...@@ -8,7 +8,7 @@ test_split: test ...@@ -8,7 +8,7 @@ test_split: test
output_type: generate_until output_type: generate_until
generation_kwargs: generation_kwargs:
num_beams: 10 num_beams: 10
max_length: 128 max_gen_toks: 128
until: until:
- "</s>" - "</s>"
doc_to_text: !function utils.doc_to_text doc_to_text: !function utils.doc_to_text
......
...@@ -8,7 +8,7 @@ test_split: test ...@@ -8,7 +8,7 @@ test_split: test
output_type: generate_until output_type: generate_until
generation_kwargs: generation_kwargs:
num_beams: 10 num_beams: 10
max_length: 128 max_gen_toks: 128
until: until:
- "</s>" - "</s>"
doc_to_text: !function utils.doc_to_text doc_to_text: !function utils.doc_to_text
......
...@@ -8,7 +8,7 @@ test_split: test ...@@ -8,7 +8,7 @@ test_split: test
output_type: generate_until output_type: generate_until
generation_kwargs: generation_kwargs:
num_beams: 10 num_beams: 10
max_length: 128 max_gen_toks: 128
until: until:
- "</s>" - "</s>"
doc_to_text: !function utils.doc_to_text doc_to_text: !function utils.doc_to_text
......
...@@ -8,7 +8,7 @@ test_split: test ...@@ -8,7 +8,7 @@ test_split: test
output_type: generate_until output_type: generate_until
generation_kwargs: generation_kwargs:
num_beams: 10 num_beams: 10
max_length: 128 max_gen_toks: 128
until: until:
- "</s>" - "</s>"
doc_to_text: !function utils.doc_to_text doc_to_text: !function utils.doc_to_text
......
...@@ -8,7 +8,7 @@ test_split: test ...@@ -8,7 +8,7 @@ test_split: test
output_type: generate_until output_type: generate_until
generation_kwargs: generation_kwargs:
num_beams: 10 num_beams: 10
max_length: 128 max_gen_toks: 128
until: until:
- "</s>" - "</s>"
doc_to_text: !function utils.doc_to_text doc_to_text: !function utils.doc_to_text
......
...@@ -8,7 +8,7 @@ test_split: test ...@@ -8,7 +8,7 @@ test_split: test
output_type: generate_until output_type: generate_until
generation_kwargs: generation_kwargs:
num_beams: 10 num_beams: 10
max_length: 128 max_gen_toks: 128
until: until:
- "</s>" - "</s>"
doc_to_text: !function utils.doc_to_text doc_to_text: !function utils.doc_to_text
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment