ModelScope中,输出的size为啥是(1,128,21128)?

"关于Structbert预训练模型(iic/nlp_structbert_backbone_base_std)的问题请教一下,
model_id = 'iic/nlp_structbert_backbone_base_std'
config = SbertConfig.from_pretrained(model_id)
tokenizer = Tokenize.from_pretrained(model_id)
pretrain = SbertModel.from_pretrained(model_id, config=config)

ModelScope中,先使用tokenizer将文本转ids,再使用pretrain(**ids)时,输出的size为啥是(1,128,21128),按理来说不应该是(batch_size,768)吗?"

展开
收起
小小爱吃香菜 2024-06-19 08:41:55 75 分享 版权
0 条回答
写回答
取消 提交回答

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

热门讨论

热门文章

还有其他疑问?
咨询AI助理