请提出你的问题:目前运行了 https://aistudio.baidu.com/projectdetail/6519985?channelType=0&channel=0,如果只是想用这模型对文本进行特征提取,该怎么做?
tnkciper1#
可以添加这个参数,将layer output作为特征输出:
https://github.com/PaddlePaddle/PaddleNLP/blob/3aa92ce26903daf2aca827356704180e833b12ab/paddlenlp/experimental/transformers/chatglm_v2/modeling.py#L479C34-L479C54
output_hidden_states=output_hidden_states,
1条答案
按热度按时间tnkciper1#
可以添加这个参数,将layer output作为特征输出:
output_hidden_states=output_hidden_states,