OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
e3609a0619
pytorch/caffe2/python
History
Alisson Gusatti Azzolini e3609a0619 Correctly propagate remap_blob across net boundaries 
Summary: If a blob is copy from device A to device B in the init_net, and then is used as an external_input in the train_net, we want the train_net to correctly use the blob already on device B instead of copying it over and over again.

Reviewed By: akyrola

Differential Revision: D5800870

fbshipit-source-id: d93f44bba80e4ed70eb03183d552496b54a966b5
2017-09-24 21:21:57 -07:00
..
docs Standardize operator argument "is_test" 2017-09-13 14:35:27 -07:00
examples for CPU-data parallel, allow sharing model 2017-09-15 16:19:37 -07:00
helpers resubmission of PR1175: fp16 BatchMatMul 2017-09-14 21:46:05 -07:00
layers Make LastNWindowCollector thread-safe 2017-09-22 09:48:30 -07:00
mint
mkl Support grouped convolutions in MKL 2017-07-25 14:19:02 -07:00
modeling Scaled training and fetching from the PS 2017-08-23 18:16:03 -07:00
models De-dup beam search state reshape shape blob 2017-09-23 23:19:44 -07:00
operator_test Adding uint8 support for to code generator for and high-performance emebding look-up kernels, supporting 2017-09-21 14:50:43 -07:00
predictor Allow specification of num_workers in PredictorExportMeta and enable for NMT beam search model 2017-09-07 22:48:45 -07:00
rnn Revert D5589309: modify _LSTM into _RNN to adapt GRU 2017-08-10 16:42:41 -07:00
_import_c_extension.py
allcompare_test.py Adding AllCompare-like function to data_parallel_model 2017-07-13 13:03:57 -07:00
attention.py resubmission of PR1175: fp16 BatchMatMul 2017-09-14 21:46:05 -07:00
benchmark_generator.py A benchmark generator for individual ops 2017-08-31 17:33:21 -07:00
binarysize.py binary size util 2017-07-14 17:49:24 -07:00
brew_test.py Allow passing unsymmetric 2d kernels to brew.conv. 2017-08-10 15:27:16 -07:00
brew.py resubmission of PR1175: fp16 BatchMatMul 2017-09-14 21:46:05 -07:00
build.py Strip Operator Schema in mobile build 2017-08-22 13:31:08 -07:00
caffe_translator_test.py Add option to remove legacy_pad in caffe_translator 2017-09-14 10:32:48 -07:00
caffe_translator.py Fix caffe translator 2017-09-19 09:21:14 -07:00
checkpoint_test.py Changes the checkpoint naming rules. 2017-08-17 22:16:42 -07:00
checkpoint.py Allow CheckpointManager init() and load() to use a different db type with path_prefix 2017-09-22 09:48:29 -07:00
CMakeLists.txt
cnn.py cnnmodelhelper deprecate warning 2017-05-18 23:35:26 -07:00
context_test.py Add default implementation of __call__ for context manager 2017-08-22 17:46:22 -07:00
context.py Add default implementation of __call__ for context manager 2017-08-22 17:46:22 -07:00
control_ops_grad.py Backpropagation for If operator 2017-09-18 16:17:42 -07:00
control_ops_util.py Backpropagation for If operator 2017-09-18 16:17:42 -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 tests: use assertRaises, not expectedFail 2017-09-13 11:39:35 -07:00
core_test.py Correctly propagate remap_blob across net boundaries 2017-09-24 21:21:57 -07:00
core.py Correctly propagate remap_blob across net boundaries 2017-09-24 21:21:57 -07:00
crf.py Deprecate CNNModelHelper in python/crf.py 2017-06-14 08:49:27 -07:00
data_parallel_model_test.py for CPU-data parallel, allow sharing model 2017-09-15 16:19:37 -07:00
data_parallel_model.py Fix typo in warning in data_parallel_model 2017-09-22 23:03:28 -07:00
data_workers_test.py Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py 2017-08-07 10:14:08 -07:00
data_workers.py Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py 2017-08-07 10:14:08 -07:00
dataio_test.py Allow tasks/execution_steps to be cloned at runtime 2017-06-20 22:32:07 -07:00
dataio.py task api, fix comments - a bit cleanup 2017-09-18 21:43:20 -07:00
dataset.py TrimDataset to ensure size is multiple of number or replicas 2017-09-13 12:17:21 -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
embedding_generation_benchmark.py Benchmark for embedding generation 2017-08-15 14:22:41 -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
functional_test.py Functional C2 2017-09-13 15:37:28 -07:00
functional.py Functional C2 2017-09-13 15:37:28 -07:00
gradient_check_test.py Backpropagation for If operator 2017-09-18 16:17:42 -07:00
gradient_checker.py Fix a few typos and grammars in comment 2017-06-14 18:22:39 -07:00
gru_cell.py Revert D5589309: modify _LSTM into _RNN to adapt GRU 2017-08-10 16:42:41 -07:00
hsm_util.py
hypothesis_test_util.py Consistent behavior of max reduction for segment ops and fix test 2017-09-20 10:59:43 -07:00
hypothesis_test.py EnsureDense/SparseToDense for CUDA 2017-09-01 09:33:05 -07:00
layer_model_helper.py Add option to ignore parameter initialization 2017-09-20 12:03:22 -07:00
layer_model_instantiator.py saving/loading CPU/GPU nets 2017-07-23 02:18:15 -07:00
layer_parameter_sharing_test.py Adding parameter sharing API to Dper2 2017-08-03 00:33:18 -07:00
layer_test_util.py Add a method to run a train net multiple times in layer_test_util.py 2017-07-28 19:56:05 -07:00
layers_test.py fix layers_test: atol should almost always accompany rtol 2017-09-22 23:31:01 -07:00
lengths_reducer_rowwise_8bit_ops_test.py Rowwise quantization 2017-09-06 10:19:38 -07:00
load_save_test.py
lstm_benchmark.py enable setting rnn executor threads and max streams 2017-09-08 16:36:51 -07:00
memonger_test.py Trying to fix all networkx 2 issues. 2017-09-21 16:01:47 -07:00
memonger.py Trying to fix all networkx 2 issues. 2017-09-21 16:01:47 -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 YellowFin GPU class and Python optimizer 2017-08-30 18:32:24 -07:00
modifier_context.py add base class ModifierContext, rewrite OptimizerContext, add RegularizerContext 2017-09-08 11:39:23 -07:00
mpi_python.cc
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 Control flow operators 2017-08-28 20:04:43 -07:00
net_builder.py Backpropagation for If operator 2017-09-18 16:17:42 -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 net_printer.to_string() accepts NetDef 2017-08-01 10:17:29 -07:00
optimizer_context.py add base class ModifierContext, rewrite OptimizerContext, add RegularizerContext 2017-09-08 11:39:23 -07:00
optimizer_test_util.py Added support for scaling learning rate of Caffe2 optimizers during training 2017-08-25 19:04:47 -07:00
optimizer_test.py Disabled test for equivalency between Caffe2's and Numpy's YellowFin 2017-09-06 13:47:45 -07:00
optimizer.py Decayed adagrad 2017-09-15 00:35:21 -07:00
parallel_workers_test.py Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py 2017-08-07 10:14:08 -07:00
parallel_workers.py Caffe2 [easy]: Better exception logging in parallel_workers/data_workers 2017-08-10 15:27:19 -07:00
parallelize_gpu_bmuf_distributed_test.py AddBlobsSync() 2017-09-12 10:33:22 -07:00
pipeline_test.py Ability to dequeue and concat multiple records in a single QueueDequeue op 2017-08-31 10:48:59 -07:00
pipeline.py task api, fix comments - a bit cleanup 2017-09-18 21:43:20 -07:00
predictor_constants.py
pybind_state_gpu.cc workspace_gpu: Get{CUDAVersion,DeviceProperties} 2017-09-17 20:01:34 -07:00
pybind_state_mkl.cc MKL code move 2017-07-26 20:21:55 -07:00
pybind_state.cc Functional C2 2017-09-13 15:37:28 -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 Ability to dequeue and concat multiple records in a single QueueDequeue op 2017-08-31 10:48:59 -07:00
record_queue.py Fix a few typos and grammars in comment 2017-06-14 18:22:39 -07:00
recurrent.py make rnn executor figure out recurrent mappings from links 2017-09-09 16:19:26 -07:00
regularizer_context.py add base class ModifierContext, rewrite OptimizerContext, add RegularizerContext 2017-09-08 11:39:23 -07:00
regularizer.py add regulariztion in caffe2 and dper 2017-09-08 11:39:22 -07:00
rnn_cell.py remove in-place Dropout from rnn_cell (bug in PR-1185) 2017-09-15 14:03:33 -07:00
schema_test.py Return empty Struct when get_field has empty input 2017-08-01 19:49:47 -07:00
schema.py Make LastNWindowCollector thread-safe 2017-09-22 09:48:30 -07:00
scope_test.py
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
session.py Adds the master setup plan to the model exporter. 2017-08-25 16:01:24 -07:00
sparse_to_dense_mask_test.py Add more enforces to SparseToDenseMask operator. 2017-09-02 02:16:24 -07:00
task.py Support session in distributed realtime trainer 2017-08-16 10:28:55 -07:00
test_util.py Clear the operator default engines before running operator tests 2017-08-29 17:47:20 -07:00
text_file_reader.py
timeout_guard.py Revert D5655753: [Caffe2] better straggler exit procedure 2017-08-25 14:23:09 -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 Update proto definition 2017-08-22 19:01:18 -07:00
visualize.py Python 3 compatible integer division 2017-07-06 11:47:12 -07:00
workspace_test.py ApplyTransformIfFaster 2017-08-17 15:36:51 -07:00
workspace.py workspace_gpu: Get{CUDAVersion,DeviceProperties} 2017-09-17 20:01:34 -07:00