- 参数加载到模型中:
from transformers import BartForConditionalGeneration, BertTokenizer
def avg_parameters(model_files, save_model=None):
avg_model = BartForConditionalGeneration.from_pretrained(model_files[0])
avg_model_p = avg_model.state_dict()
parm_name = []
for i in avg_model.named_parameters():
parm_name.append(i[0])
mode