torchcrf库

在进行命名实体识别时，使用了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)
# 深度学习 # 人工智能
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

VLA-LPAF：视觉-语言-动作模型的轻量级视角-自适应融合，实现更不受约束的机器人操作

avatar 2048 AI社区

Laravel与其他编程语言的比较

因此，在比较时，我们将 Laravel 与其他编程语言的流行框架进行对比，如 Python 的 Django、Ruby 的 Ruby on Rails、JavaScript 的 Express.js（基于 Node.js），以及 Java 的 Spring Boot。Express.js (JavaScript/Node.js): JavaScript 语法灵活，Express.js 使用中间件模

avatar 2048 AI社区
cover

多智能体系统框架：分层设计与高效执行的新范式

avatar 2048 AI社区