pytorch/caffe2/python
Kittipat Virochsiri 22d4eaeb9e JoinContext
Summary:
Layer to allow model to follow different paths for each instantiation context and join later. Together with tagging system cleanup (this is a separate issue), this should reduce the need to write a layer to differentiate between context.

Re: tagging system clean up, we should make exclusion more explicit: EXCLUDE_FROM_<CONTEXT>. This would simplify instation code. TRAIN_ONLY should become a set of all EXCLUDE_FROM_*, except EXCLUDE_FROM_TRAIN.

Reviewed By: kennyhorror

Differential Revision: D4964949

fbshipit-source-id: ba6453b0deb92d1989404efb9d86e1ed25297202
2017-05-02 17:32:26 -07:00
..
docs
examples resnet train print loss and accuracy 2017-04-25 16:03:58 -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
mkl MKL related files with review comments incorporated 2017-04-25 00:31:29 -07:00
models Move BeamSearchForwardOnly to OSS 2017-04-27 13:35:53 -07:00
operator_test caffe2: rebatching queue for MultiTask 2017-05-02 15:22:46 -07:00
predictor Re-apply #266 2017-04-25 21:17:04 -07:00
_import_c_extension.py
attention.py unbreak test_seq2seq_caffe2_model_cnn_one_stack_encoder 2017-04-20 10:06:25 -07:00
brew.py Adding add_weight_decay and image_input to brew module 2017-04-25 16:03:58 -07:00
brew_test.py brew fc test fix for packed_fc 2017-05-01 14:33:05 -07:00
caffe_translator.py Add Reduction layer in caffe_translator 2017-04-07 16:17:07 -07:00
caffe_translator_test.py
checkpoint.py Adds interfaces to check the existence of a DB 2017-04-11 14:07:49 -07:00
checkpoint_test.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.py
context_test.py
control.py
control_test.py
convnet_benchmarks.py
convnet_benchmarks_test.py
core.py JoinContext 2017-05-02 17:32:26 -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
crf.py Integrate CRF in DeepText + New caffe2 operator for viterbi decode 2017-05-01 20:39:41 -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_parallel_model_test.py RNNCell, LSTMCell, LSTMWithAttentionCell 2017-04-18 00:47:20 -07:00
data_workers.py Port Xray OC workflow to elastic_data_parallel_model 2017-05-01 00:32:47 -07:00
data_workers_test.py Allow stopping of specific data workers + specify c2 queue size 2017-04-26 11:18:40 -07:00
dataio.py
dataio_test.py
dataset.py
db_test.py
device_checker.py
dyndep.py
experiment_util.py Port Xray OC workflow to elastic_data_parallel_model 2017-05-01 00:32:47 -07:00
extension_loader.py
gradient_check_test.py
gradient_checker.py Ensure sparse_gradients feed to CPU 2017-05-02 15:01:26 -07:00
hsm_util.py
hypothesis_test.py NormalizeOP CUDA impelementation 2017-05-01 09:25:30 -07:00
hypothesis_test_util.py Add option to control the size of lengths tensor 2017-04-20 09:53:22 -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: support dropping hidden / cell states when sequence 2017-04-27 11:47:29 -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
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
mkl_test_util.py
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
muji.py
muji_test.py
net_builder.py
net_builder_test.py
net_drawer.py
net_printer.py
net_printer_test.py
optimizer.py Returns auxiliary parameters in the optimizers. 2017-04-17 10:16:32 -07:00
optimizer_test.py Returns auxiliary parameters in the optimizers. 2017-04-17 10:16:32 -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
pipeline.py
predictor_constants.py Re-apply #266 2017-04-25 21:17:04 -07:00
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
pybind_state_gpu.cc
pybind_state_mkl.cc
python_op_test.py
queue_util.py
record_queue.py
recurrent.py special executor class for RecurrentNetworks (just single threaded now) 2017-05-01 19:06:25 -07:00
rnn_cell.py Layered LSTM (naive version) 2017-04-27 19:16:58 -07:00
schema.py make debug message more explicit & verbose 2017-05-02 12:39:14 -07:00
schema_test.py fix getting empty struct 2017-04-19 22:36:05 -07:00
scope.py Fix corruption of NameScope when exception is thrown 2017-04-24 22:46:27 -07:00
scope_test.py Fix corruption of NameScope when exception is thrown 2017-04-24 22:46:27 -07:00
session.py
session_test.py Warn on setting blob on Scalar 2017-05-01 20:18:30 -07:00
sparse_to_dense_mask_test.py
task.py
test_util.py
text_file_reader.py
timeout_guard.py
toy_regression_test.py
tt_core.py
tt_core_test.py
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
workspace.py create_net: explicitly specify if one wants to overwrite the network. 2017-04-17 21:46:53 -07:00
workspace_test.py create_net: explicitly specify if one wants to overwrite the network. 2017-04-17 21:46:53 -07:00