search

kmkurn / pytorch-crf

(Linear-chain) Conditional random field in PyTorch.
https://pytorch-crf.readthedocs.io
MIT License
943 stars 152 forks source link

Size of the model with pytorch-crf. #35

Closed cinjon closed 5 years ago

cinjon commented 5 years ago

After adding in pytorch-crf, I am running into issues with the size of the model on the GPU. Is it expected to balloon a lot?

kmkurn commented 5 years ago

The memory complexity should be O(n^2) where n is the number of tags. This may be a problem if you have a high number of tags.

cinjon commented 5 years ago

thanks.