However, when I add custom task, it will output
TypeError: __init__() got an unexpected keyword argument 'model_name'
which seems that huggingface transformers 'BertForTokenClassification' class doesn't have these attributes. I am wondering why this doesn't happen in your experiments.
This might be a stupid question, but I saw in the utils folder, it has this line:
self.model = BertForTokenClassification(model_name=current_name, model_path=target_model)
However, when I add custom task, it will output
TypeError: __init__() got an unexpected keyword argument 'model_name'
which seems that huggingface transformers 'BertForTokenClassification' class doesn't have these attributes. I am wondering why this doesn't happen in your experiments.
nevermind. I saw the overriding class. Closed