Error using CRF

[ys7yoo]

[/content/korner/model/net.py](https://h2vdoxku0l4-496ff2e9c6d22116-0-colab.googleusercontent.com/outputframe.html?vrz=colab-20230109-060104-RC00_500662758#) in __init__(self, config, num_classes, vocab)
     34         self.dropout = nn.Dropout(config.dropout)
     35         self.position_wise_ff = nn.Linear(config.hidden_size, num_classes)
---> 36         self.crf = CRF(num_labels=num_classes)
     37 
     38     def forward(self, input_ids, token_type_ids=None, tags=None):

TypeError: __init__() got an unexpected keyword argument 'num_labels'

[ys7yoo]

https://pytorch-crf.readthedocs.io/en/stable/