it seems that model dosen't have the "albert.pooler.weight" parameter
#6
by
rileyzhang
- opened
When executing the command
"python convert_albert_from_huggingface_to_uer.py --input_model_path xxx --output_model_path xxx"
an error was reported:
Traceback (most recent call last):
File "C:/Users/rileyzhang/PycharmProjects/UER-py/scripts/convert_albert_from_huggingface_to_uer.py", line 67, in
output_model["target.sp.linear_1.weight"] = input_model["albert.pooler.weight"]
KeyError: 'albert.pooler.weight'
I print the total parameter of the model, and do not find the parameter or parameters with similar name.
Could it be that the model has not been pre-trained for the SOP task?