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={
|