Tao Wu 6530db49bc improve pair_wise_loss operator to support multiple sessions ...

Summary: The diff adds support for rank_loss operator to support computing loss for multiple sessions (batch).

Reviewed By: kittipatv

Differential Revision: D5515465

fbshipit-source-id: 55a01cd5ad21eaeae82875ad136c392fed0dbb26

2017-07-28 15:12:47 -07:00

..

docs

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

examples

warn about using test/val model with init_params=True + fixed some cases

2017-07-27 13:20:27 -07:00

helpers

Adding tanh to brew

2017-07-11 18:17:52 -07:00

layers

Fixing semi-random layer model for multi-layer models

2017-07-27 15:25:19 -07:00

mint

doxygen python block added

2017-03-29 06:46:16 -07:00

mkl

Support grouped convolutions in MKL

2017-07-25 14:19:02 -07:00

modeling

allow param_info to set optimizer

2017-07-12 08:49:48 -07:00

models

Fix broken seq2seq example

2017-07-13 23:31:54 -07:00

operator_test

improve pair_wise_loss operator to support multiple sessions

2017-07-28 15:12:47 -07:00

predictor

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

rnn

Moved sigmoid, tanh, and _prepare_lstm (renamed) to a util file.

2017-07-10 17:52:22 -07:00

_import_c_extension.py

doxygen python block added

2017-03-29 06:46:16 -07:00

allcompare_test.py

Adding AllCompare-like function to data_parallel_model

2017-07-13 13:03:57 -07:00

attention.py

attention with encoder_lengths

2017-07-23 10:06:01 -07:00

binarysize.py

binary size util

2017-07-14 17:49:24 -07:00

brew_test.py

quick fix for model_helper __init__

2017-07-12 08:49:48 -07:00

brew.py

Adding tanh to brew

2017-07-11 18:17:52 -07:00

caffe_translator_test.py

…

caffe_translator.py

Read pretrained weights using binary mode in caffe_translator.py

2017-07-08 10:17:57 -07:00

checkpoint_test.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

checkpoint.py

Adds interfaces to check the existence of a DB

2017-04-11 14:07:49 -07:00

CMakeLists.txt

…

cnn.py

cnnmodelhelper deprecate warning

2017-05-18 23:35:26 -07:00

context_test.py

…

context.py

doxygen python block added

2017-03-29 06:46:16 -07:00

control_test.py

…

control.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

convnet_benchmarks_test.py

…

convnet_benchmarks.py

brew API in convnet benchmark

2017-07-05 10:34:48 -07:00

core_gradients_test.py

warn about orphan StopGradient output

2017-07-20 21:41:41 -07:00

core_test.py

quick fix inplace blob bug

2017-07-23 02:18:16 -07:00

core.py

quick fix inplace blob bug

2017-07-23 02:18:16 -07:00

crf.py

Deprecate CNNModelHelper in python/crf.py

2017-06-14 08:49:27 -07:00

data_parallel_model_test.py

warn about using test/val model with init_params=True + fixed some cases

2017-07-27 13:20:27 -07:00

data_parallel_model.py

warn about using test/val model with init_params=True + fixed some cases

2017-07-27 13:20:27 -07:00

data_workers_test.py

fix a rare race condition by initializing scratch blobs beforehand

2017-06-26 10:18:18 -07:00

data_workers.py

add timeout argument to DequeueBlobs; use 10 min timeout for data workers

2017-07-13 18:52:03 -07:00

dataio_test.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

dataio.py

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -07:00

dataset.py

Use scope name for dataset cursor

2017-07-15 19:22:32 -07:00

db_test.py

String-related fixes for Python 3

2017-05-26 16:04:32 -07:00

device_checker.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

dyndep.py

doxygen python block added

2017-03-29 06:46:16 -07:00

empty.so

Adding video data layer for caffe2

2017-05-05 14:16:38 -07:00

experiment_util.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

extension_loader.py

Make extension loader properly handle visibility.

2017-03-30 14:38:38 -07:00

gradient_check_test.py

Cos, Sin, and Abs operators

2017-07-03 22:18:32 -07:00

gradient_checker.py

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -07:00

gru_cell.py

Implemented GRUCell

2017-07-10 17:52:25 -07:00

hsm_util.py

doxygen python block added

2017-03-29 06:46:16 -07:00

hypothesis_test_util.py

Add support for int32 indices in SparseLengthSum and friends

2017-07-24 17:50:00 -07:00

hypothesis_test.py

distance_op_test from hypothesis_test refactored

2017-07-26 13:37:08 -07:00

layer_model_helper.py

Fixing error message for layer model helper

2017-07-18 09:52:45 -07:00

layer_model_instantiator.py

saving/loading CPU/GPU nets

2017-07-23 02:18:15 -07:00

layer_test_util.py

Core unit test fixes for Python 3

2017-06-23 13:22:16 -07:00

layers_test.py

Fixing semi-random layer model for multi-layer models

2017-07-27 15:25:19 -07:00

load_save_test.py

Allow Load operator to load into overriden names

2017-04-27 01:18:12 -07:00

lstm_benchmark.py

Added flags to lstm, convnet and sparse_nn_benchmarks to print out operators

2017-06-30 23:47:04 -07:00

memonger_test.py

fix for duplicate input case

2017-07-13 01:51:30 -07:00

memonger.py

add code comments to memonger

2017-07-17 13:07:33 -07:00

mkl_test_util.py

Implement a filler op test

2017-07-25 14:18:57 -07:00

model_device_test.py

Deprecate CNNModelHelper in caffe2/python/model_device_test.py

2017-06-22 15:37:17 -07:00

model_helper.py

ExtractPredictorNet should strip gpu_id prefix from step_net

2017-07-27 16:06:47 -07:00

mpi_python.cc

Fix pybind11 module name for MPI helpers

2017-05-02 23:18:50 -07:00

muji_test.py

Fixes range/xrange for Python 3

2017-06-07 00:04:26 -07:00

muji.py

Fixes range/xrange for Python 3

2017-06-07 00:04:26 -07:00

net_builder_test.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

net_builder.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

net_drawer.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

net_printer_test.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

net_printer.py

Fix net_printer.py

2017-07-11 15:26:52 -07:00

optimizer_context.py

allow param_info to set optimizer

2017-07-12 08:49:48 -07:00

optimizer_test_util.py

Fp16 training initializers

2017-06-01 08:34:46 -07:00

optimizer_test.py

Fix optimizer_context OSS test

2017-07-20 12:21:09 -07:00

optimizer.py

Populate learning rate blob name into data_parallel_model and fix resnet50_trainer example.

2017-07-21 16:24:10 -07:00

parallelize_gpu_bmuf_distributed_test.py

Add distributed BMUF implementation.

2017-06-21 16:18:11 -07:00

pipeline.py

Enable runtime cloning of tasks.

2017-06-21 03:18:20 -07:00

predictor_constants.py

Re-apply #266

2017-04-25 21:17:04 -07:00

pybind_state_gpu.cc

…

pybind_state_mkl.cc

MKL code move

2017-07-26 20:21:55 -07:00

pybind_state.cc

MKL code move

2017-07-26 20:21:55 -07:00

pybind_state.h

fast simple-net memonger for C++

2017-07-06 15:17:07 -07:00

python_op_test.py

Fix some typos

2017-06-28 13:50:48 -07:00

queue_util.py

doxygen python block added

2017-03-29 06:46:16 -07:00

record_queue.py

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -07:00

recurrent.py

RNN Workspace Blob Extraction

2017-07-17 10:24:18 -07:00

rnn_cell.py

use model.create_param instead of using param_init_net directly

2017-07-24 21:17:24 -07:00

schema_test.py

Add __sub__ function for schema.Struct

2017-06-28 11:24:01 -07:00

schema.py

IndexHash

2017-07-07 23:06:11 -07:00

scope_test.py

Fix corruption of NameScope when exception is thrown

2017-04-24 22:46:27 -07:00

scope.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

session_test.py

Warn on setting blob on Scalar

2017-05-01 20:18:30 -07:00

session.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

sparse_to_dense_mask_test.py

String-related fixes for Python 3

2017-05-26 16:04:32 -07:00

task.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

test_util.py

doxygen python block added

2017-03-29 06:46:16 -07:00

text_file_reader.py

doxygen python block added

2017-03-29 06:46:16 -07:00

timeout_guard.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

toy_regression_test.py

…

tt_core_test.py

…

tt_core.py

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -07:00

utils.py

Support (U)INT8, (U)INT16 in data type conversion

2017-07-26 11:23:53 -07:00

visualize.py

Python 3 compatible integer division

2017-07-06 11:47:12 -07:00

workspace_test.py

Core unit test fixes for Python 3

2017-06-23 13:22:16 -07:00

workspace.py

Caffe2: don't swallow exception stacktrace

2017-07-26 15:48:05 -07:00