input features
Browse files
tasks.py
CHANGED
@@ -134,7 +134,7 @@ TaskRegistry.add(
|
|
134 |
preprocessors.span_corruption,
|
135 |
seqio.preprocessors.append_eos_after_trim,
|
136 |
],
|
137 |
-
output_features={"targets": DEFAULT_OUTPUT_FEATURES["targets"]},
|
138 |
metric_fns=[]
|
139 |
)
|
140 |
|
@@ -216,7 +216,7 @@ TaskRegistry.add(
|
|
216 |
preprocessors.prefix_lm,
|
217 |
seqio.preprocessors.append_eos_after_trim,
|
218 |
],
|
219 |
-
output_features=
|
220 |
metric_fns=[]
|
221 |
)
|
222 |
|
|
|
134 |
preprocessors.span_corruption,
|
135 |
seqio.preprocessors.append_eos_after_trim,
|
136 |
],
|
137 |
+
output_features={"targets": DEFAULT_OUTPUT_FEATURES["targets"],"inputs": seqio.Feature(add_eos=True)},
|
138 |
metric_fns=[]
|
139 |
)
|
140 |
|
|
|
216 |
preprocessors.prefix_lm,
|
217 |
seqio.preprocessors.append_eos_after_trim,
|
218 |
],
|
219 |
+
output_features={"targets": DEFAULT_OUTPUT_FEATURES["targets"]},
|
220 |
metric_fns=[]
|
221 |
)
|
222 |
|