TorchCRF库出现‘mask of the first timestep must all be on‘错误

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)