saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/python
History
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
..
docs Documenation generation to wiki 2017-02-15 16:00:44 -08:00
examples Remove more instances of batch_size 2017-03-19 22:31:30 -07:00
layers allows to specify output names for functional layers 2017-03-23 13:18:58 -07:00
mint
models exec_net --> predict_net 2017-03-23 16:31:49 -07:00
operator_test CRF layer in caffe2 2017-03-23 22:02:02 -07:00
_import_c_extension.py
attention.py Removing unnecessary Copy after fixing gradients for external parameters 2017-03-23 15:04:22 -07:00
caffe_translator.py translator fix 2017-03-22 15:09:13 -07:00
caffe_translator_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
checkpoint.py Fix issues pickling jobs 2017-02-21 20:47:27 -08:00
checkpoint_test.py Fix issues pickling jobs 2017-02-21 20:47:27 -08:00
CMakeLists.txt
cnn.py Add ConvNd to model helper 2017-03-22 15:47:48 -07:00
context.py Make ContextManager thread-safe 2017-02-13 19:45:35 -08:00
context_test.py Make ContextManager thread-safe 2017-02-13 19:45:35 -08:00
control.py
control_test.py
convnet_benchmarks.py Convnet benchmark cudnn_ws 2017-03-02 15:32:37 -08:00
convnet_benchmarks_test.py
core.py fixes to make data parallel model work for RecurrentNet + test case 2017-03-14 15:48:07 -07:00
core_gradients_test.py add inference for gradient ops + a couple of missing shape inference functions + fix to scalars 2017-02-28 23:33:32 -08:00
core_test.py NextScopedBlob with well-defined behavior and respect namescope 2017-02-16 17:16:36 -08:00
crf.py CRF layer in caffe2 2017-03-23 22:02:02 -07:00
data_parallel_model.py add gpu support for caffe2-seq2seq 2017-03-17 05:19:14 -07:00
data_parallel_model_test.py fixes to make data parallel model work for RecurrentNet + test case 2017-03-14 15:48:07 -07:00
data_workers.py Remove use of logging module and np.random.randint() due to deadlocks with forks 2017-03-01 03:32:56 -08:00
data_workers_test.py close blobs queues when stopping + test 2017-02-27 10:07:57 -08:00
dataio.py Stop multi_reader if we run out of data before max_examples 2017-03-10 18:03:57 -08:00
dataio_test.py Stop multi_reader if we run out of data before max_examples 2017-03-10 18:03:57 -08:00
dataset.py
db_test.py
device_checker.py
dyndep.py
experiment_util.py fix external_loggers 2017-03-16 13:47:58 -07:00
extension_loader.py
gradient_check_test.py
gradient_checker.py
hsm_util.py
hypothesis_test.py add AccumulateHistogramOp 2017-03-08 19:37:32 -08:00
hypothesis_test_util.py Allow use of ReversePackedSegs operator in CUDA context 2017-03-09 15:03:55 -08:00
introspect_vis.py User input (Conv out, etc.) 2017-03-08 13:49:45 -08:00
layer_model_helper.py Use new metric intefaces in trainer workflows. 2017-03-07 12:46:52 -08:00
layer_model_instantiator.py Migrate realtime training workflows to use new metrics. 2017-03-08 23:49:41 -08:00
layers_test.py allows to specify output names for functional layers 2017-03-23 13:18:58 -07:00
load_save_test.py Improve error message from LogFileDB on missing file 2017-03-10 23:31:28 -08:00
lstm_benchmark.py LSTM benchmark (Caffe2 RNN based) 2017-02-28 23:17:26 -08:00
memonger.py
memonger_test.py
mkl_test_util.py
model_device_test.py
model_helper.py add gpu support for caffe2-seq2seq 2017-03-17 05:19:14 -07:00
mpi_python.cc
muji.py
muji_test.py
net_builder.py Improve "reporter net" design 2017-02-21 20:17:40 -08:00
net_builder_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
net_drawer.py Add model graph to dper_example 2017-02-07 13:03:54 -08:00
net_printer.py Add task outputs and stop signals to net_printer 2017-03-07 01:21:40 -08:00
net_printer_test.py Debug/Analysis tools for Jobs/ExecutionSteps 2017-02-06 17:31:20 -08:00
optimizer.py refactor and modulize optimizers 2017-03-07 18:46:47 -08:00
optimizer_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
optimizer_test_util.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
pipeline.py Better names for nets, steps and tasks 2017-02-09 16:33:54 -08:00
pybind_state.cc Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00
pybind_state.h bugfix for Windows, esp. VS 2017 2017-03-21 05:17:59 -07:00
pybind_state_gpu.cc Cudnn v6 2017-02-28 17:46:33 -08:00
pybind_state_mkl.cc
python_op_test.py
queue_util.py Better names for nets, steps and tasks 2017-02-09 16:33:54 -08:00
record_queue.py
recurrent.py Removing unnecessary Copy after fixing gradients for external parameters 2017-03-23 15:04:22 -07:00
schema.py migrate mtml to dper2 2017-03-16 17:48:05 -07:00
schema_test.py support get/add a field by nested name 2017-03-15 02:00:57 -07:00
scope.py
scope_test.py
session.py Default LocalSession to current workspace. 2017-03-01 16:03:18 -08:00
session_test.py NextScopedBlob with well-defined behavior and respect namescope 2017-02-16 17:16:36 -08:00
sparse_to_dense_mask_test.py
task.py Gather perf counters for distributed jobs 2017-02-21 22:06:25 -08:00
test_util.py
text_file_reader.py fix typo in TextFileReader 2017-02-21 14:02:48 -08:00
timeout_guard.py Euthanize a process with timeout 2017-03-01 11:38:11 -08:00
toy_regression_test.py
tt_core.py
tt_core_test.py
utils.py Add a create your own dataset tutorial 2017-02-22 03:31:47 -08:00
visualize.py
workspace.py Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00
workspace_test.py Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00