Aapo Kyrola 1a831ce8f2 Add direct enqueuing to enable RNN input, allow specify batch columns ...

Summary:
Add a parameter dont_rebatch to data_workers. This disables batching of input from fetcher to equal-batch size chunks. This is not desired with RNNs where with longer sequence length we might want to have smaller batches etc.

For some reason the graceful-shutdown test interfered with other tests, so I removed it.

Reviewed By: jay-mahadeokar

Differential Revision: D4988549

fbshipit-source-id: cbab46d77c948f2e293e79e6eb538dde17d800ee

2017-05-03 14:49:44 -07:00

..

docs

doxygen python block added

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

examples

Inference code for seq2seq model

2017-05-02 22:47:28 -07:00

helpers

fix LRN order

2017-04-27 16:46:47 -07:00

layers

JoinContext

2017-05-02 17:32:26 -07:00

mint

doxygen python block added

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

mkl

MKL related files with review comments incorporated

2017-04-25 00:31:29 -07:00

models

Inference code for seq2seq model

2017-05-02 22:47:28 -07:00

operator_test

Add ScatterWeightedSum for GPU.

2017-05-03 13:40:48 -07:00

predictor

Re-apply #266

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

_import_c_extension.py

doxygen python block added

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

attention.py

unbreak test_seq2seq_caffe2_model_cnn_one_stack_encoder

2017-04-20 10:06:25 -07:00

brew_test.py

brew fc test fix for packed_fc

2017-05-01 14:33:05 -07:00

brew.py

Adding add_weight_decay and image_input to brew module

2017-04-25 16:03:58 -07:00

caffe_translator_test.py

Allow test discovery in caffe2/python/

2017-03-14 18:16:41 -07:00

caffe_translator.py

Add Reduction layer in caffe_translator

2017-04-07 16:17:07 -07:00

checkpoint_test.py

Adds interfaces to check the existence of a DB

2017-04-11 14:07:49 -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

fix LRN order

2017-04-27 16:46:47 -07:00

context_test.py

…

context.py

doxygen python block added

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

control_test.py

…

control.py

doxygen python block added

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

convnet_benchmarks_test.py

…

convnet_benchmarks.py

doxygen python block added

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

core_gradients_test.py

Fix backward pass computation when an input is used in a Fill-op input for shape

2017-04-11 19:32:22 -07:00

core_test.py

Handle net.external_inputs correctly in AppendNet

2017-05-02 11:20:26 -07:00

core.py

JoinContext

2017-05-02 17:32:26 -07:00

crf.py

Integrate CRF in DeepText + New caffe2 operator for viterbi decode

2017-05-01 20:39:41 -07:00

data_parallel_model_test.py

RNNCell, LSTMCell, LSTMWithAttentionCell

2017-04-18 00:47:20 -07:00

data_parallel_model.py

Data parallel model: Disable NCCL by default to hopefully reduce deadlocks

2017-05-02 16:09:17 -07:00

data_workers_test.py

Add direct enqueuing to enable RNN input, allow specify batch columns

2017-05-03 14:49:44 -07:00

data_workers.py

Add direct enqueuing to enable RNN input, allow specify batch columns

2017-05-03 14:49:44 -07:00

dataio_test.py

Stop multi_reader if we run out of data before max_examples

2017-03-10 18:03:57 -08:00

dataio.py

doxygen python block added

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

dataset.py

doxygen python block added

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

db_test.py

…

device_checker.py

doxygen python block added

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

dyndep.py

doxygen python block added

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

experiment_util.py

Port Xray OC workflow to elastic_data_parallel_model

2017-05-01 00:32:47 -07:00

extension_loader.py

Make extension loader properly handle visibility.

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

gradient_check_test.py

gradient checker for nets

2017-03-28 13:03:14 -07:00

gradient_checker.py

MultiRNNCell (Caffe2)

2017-05-03 10:02:31 -07:00

hsm_util.py

doxygen python block added

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

hypothesis_test_util.py

Add option to control the size of lengths tensor

2017-04-20 09:53:22 -07:00

hypothesis_test.py

NormalizeOP CUDA impelementation

2017-05-01 09:25:30 -07:00

layer_model_helper.py

Allow LayerModelHelper to keep input blobs from schema

2017-05-01 21:31:36 -07:00

layer_model_instantiator.py

JoinContext

2017-05-02 17:32:26 -07:00

layer_test_util.py

Re-apply #266

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

layers_test.py

JoinContext

2017-05-02 17:32:26 -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

lstm_benchmark: add warm-up stage, support layers

2017-05-02 20:34:00 -07:00

memonger_test.py

share forward activation blobs + pass unused free blobs down all branches + use shape infernece

2017-04-25 14:23:25 -07:00

memonger.py

share forward activation blobs + pass unused free blobs down all branches + use shape infernece

2017-04-25 14:23:25 -07:00

mkl_test_util.py

doxygen python block added

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

model_device_test.py

…

model_helper.py

Replaces the non-existing _param_init_net net by raising an exception.

2017-04-26 10:35:45 -07:00

mpi_python.cc

Fix pybind11 module name for MPI helpers

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

muji_test.py

…

muji.py

doxygen python block added

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

net_builder_test.py

Allow test discovery in caffe2/python/

2017-03-14 18:16:41 -07:00

net_builder.py

doxygen python block added

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

net_drawer.py

doxygen python block added

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

net_printer_test.py

…

net_printer.py

doxygen python block added

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

optimizer_test_util.py

create_net: explicitly specify if one wants to overwrite the network.

2017-04-17 21:46:53 -07:00

optimizer_test.py

Returns auxiliary parameters in the optimizers.

2017-04-17 10:16:32 -07:00

optimizer.py

Returns auxiliary parameters in the optimizers.

2017-04-17 10:16:32 -07:00

pipeline.py

doxygen python block added

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

predictor_constants.py

Re-apply #266

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

pybind_state_gpu.cc

Cudnn v6

2017-02-28 17:46:33 -08:00

pybind_state_mkl.cc

…

pybind_state.cc

create_net: explicitly specify if one wants to overwrite the network.

2017-04-17 21:46:53 -07:00

pybind_state.h

bugfix for Windows, esp. VS 2017

2017-03-21 05:17:59 -07:00

python_op_test.py

…

queue_util.py

doxygen python block added

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

record_queue.py

doxygen python block added

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

recurrent.py

special executor class for RecurrentNetworks (just single threaded now)

2017-05-01 19:06:25 -07:00

rnn_cell.py

MultiRNNCell (Caffe2)

2017-05-03 10:02:31 -07:00

schema_test.py

Prevent false overwriting of a field

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

schema.py

Prevent false overwriting of a field

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

scope_test.py

Fix corruption of NameScope when exception is thrown

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

scope.py

Fix corruption of NameScope when exception is thrown

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

session_test.py

Warn on setting blob on Scalar

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

session.py

doxygen python block added

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

sparse_to_dense_mask_test.py

…

task.py

doxygen python block added

2017-03-29 06:46:16 -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

doxygen python block added

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

toy_regression_test.py

…

tt_core_test.py

…

tt_core.py

doxygen python block added

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

utils.py

Free scratch blobs when data workers exits, add utility function to reset blobs

2017-04-26 13:40:13 -07:00

visualize.py

doxygen python block added

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

workspace_test.py

create_net: explicitly specify if one wants to overwrite the network.

2017-04-17 21:46:53 -07:00

workspace.py

create_net: explicitly specify if one wants to overwrite the network.

2017-04-17 21:46:53 -07:00