请提出你的问题：目前运行了 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,