def __init__(self, config: configure_finetuning.FinetuningConfig, tasks):
self._config = config
self._tasks = tasks
self._name_to_task = {task.name: task for task in tasks}
self._feature_specs = feature_spec.get_shared_feature_specs(config)
for task in tasks:
self._feature_specs += task.get_feature_specs()
self._name_to_feature_config = {
spec.name: spec.get_parsing_spec()
for spec in self._feature_specs
}
assert len(self._name_to_feature_config) == len(self._feature_specs)
def __init__(self, config: configure_finetuning.FinetuningConfig, tasks):
self._config = config
self._tasks = tasks
self._name_to_task = {task.name: task for task in tasks}
self._feature_specs = []
# multi-choice mrc overwrites input_ids, input_mask, etc.
if not any([isinstance(x, qa_tasks.MQATask) for x in tasks]):
self._feature_specs = feature_spec.get_shared_feature_specs(config)
for task in tasks:
self._feature_specs += task.get_feature_specs()
self._name_to_feature_config = {
spec.name: spec.get_parsing_spec()
for spec in self._feature_specs
}
assert len(self._name_to_feature_config) == len(self._feature_specs)
