Ahmed Aly
|
99bfd36a04
|
CRF layer in caffe2
Summary:
This is implementation of a CRF layer in caffe2 according to this paper: https://arxiv.org/abs/1603.01360
Currently this implementation works only for batch_size = 1
Reference implementations:
- Tensorflow:
63a21e0540/tensorflow/contrib/crf/python/ops/crf.py
- Theano:
https://github.com/glample/tagger/blob/master/model.py#L286
Differential Revision: D4644004
fbshipit-source-id: bf0801fd8562d11dca3fefe371c3d85e1dd69ccc
|
2017-03-23 22:02:02 -07:00 |
|