I’m fine-tuning a BERT model to perform a rather complex task. It is supposed identify targets within the input string, perform regression on each of the targets and output
- the start/end index of each identified target in the input
- the regression values for each target
in a JSON format.
From my understanding, the training itself is pretty simple but to get the output in the desired format (multiple outputs based on predicted targets in a JSON format), I need to customize the model’s head.
So, i’m not really looking for a solution here but rather resources (tutorials, papers, videos) about how to do that. Thank you!