登录社区云,与社区用户共同成长
邀请您加入社区
在进行命名实体识别时,使用了torchcrf作为最后一层的输出,出现错误’mask of the first timestep must all be on’。 原因是掩码mask第一位出现了0,要求是为1。 出现这个是因为不同版本的torchcrf输入维度都统。当输入维度为(tag_nums,batch_size)时,需要对模型实例化时输入batch_size=True
model =CRF(num_labels,batch_first=True)
有“AI”的1024 = 2048,欢迎大家加入2048 AI社区
更多推荐
Arm的“造芯”阳谋:智能体时代,CPU重新回到舞台中央
业务签约双录怎么选?避开行业普遍存在的四大选型误区
Anthropic 蒸馏攻击报告拆解:2 亿次交互背后的模型能力外流路径
扫一扫分享内容
为遵守国家网络实名制规定,未绑定将限制内容发布与互动
所有评论(0)