pere commited on
Commit
b957a31
1 Parent(s): c74aee2
Files changed (1) hide show
  1. tasks.py +1 -1
tasks.py CHANGED
@@ -167,7 +167,7 @@ seqio.TaskRegistry.add(
167
  seqio.preprocessors.tokenize_and_append_eos,
168
  ],
169
  metric_fns=[metrics.accuracy,my_metrics.f1_macro],
170
- output_features=DEFAULT_OUTPUT_FEATURES,
171
  )
172
 
173
 
 
167
  seqio.preprocessors.tokenize_and_append_eos,
168
  ],
169
  metric_fns=[metrics.accuracy,my_metrics.f1_macro],
170
+ output_features=MT5_OUTPUT_FEATURES,
171
  )
172
 
173