Fix data preparation
Browse files- src/run_ed_recipe_nlg.py +4 -0
src/run_ed_recipe_nlg.py
CHANGED
@@ -134,6 +134,10 @@ class DataTrainingArguments:
|
|
134 |
default=None,
|
135 |
metadata={"help": "An optional input evaluation data file to evaluate the perplexity on (a text file)."},
|
136 |
)
|
|
|
|
|
|
|
|
|
137 |
max_source_length: Optional[int] = field(
|
138 |
default=128,
|
139 |
metadata={
|
|
|
134 |
default=None,
|
135 |
metadata={"help": "An optional input evaluation data file to evaluate the perplexity on (a text file)."},
|
136 |
)
|
137 |
+
test_file: Optional[str] = field(
|
138 |
+
default=None,
|
139 |
+
metadata={"help": "An optional input evaluation data file to predict the perplexity on (a text file)."},
|
140 |
+
)
|
141 |
max_source_length: Optional[int] = field(
|
142 |
default=128,
|
143 |
metadata={
|