pere
/

pk-nb-t5x

pere commited on May 8, 2022

Commit

f431aa4

•

1 Parent(s): 0c9ac8a

output features

Files changed (1) hide show

tasks.py CHANGED Viewed

@@ -134,7 +134,7 @@ TaskRegistry.add(
         preprocessors.span_corruption,
         seqio.preprocessors.append_eos_after_trim,
     ],
-    output_features={"targets": DEFAULT_OUTPUT_FEATURES["targets"],"inputs": seqio.Feature(add_eos=True)},
     metric_fns=[]
 )

         preprocessors.span_corruption,
         seqio.preprocessors.append_eos_after_trim,
     ],
+    output_features={"targets": DEFAULT_OUTPUT_FEATURES["targets"],"inputs": seqio.Feature(vocabulary=vocabulary,add_eos=True)},
     metric_fns=[]
 )