search

harvardnlp / pytorch-struct

Fast, general, and tested differentiable structured prediction in PyTorch
http://harvardnlp.github.io/pytorch-struct
MIT License
1.1k stars 92 forks source link

gumbel crf #102

Open chijames opened 2 years ago

chijames commented 2 years ago

Hi,

Is the gumbel_crf function ready to use? If so, can you point me to the relevant documentation? (I cannot find it anywhere in here)

Thanks!

srush commented 2 years ago

Sorry for the delay. Yes that function works and lets you backprop through a relaxed version of any of the models you might construct. It implements the method from https://papers.nips.cc/paper/2020/hash/ea119a40c1592979f51819b0bd38d39d-Abstract.html