pytorch/caffe2/python
Artem Volkhin 0c03c8fca5 Add name_overrides argument to SaveOp
Summary:
In current implementation of SaveOp we always use names for blobs from the
current workspace. But there is a use case for replacing names in saved model:
for example, to use half-floats in prediction model but keep full-floats for
training model we might want to save a blob "w_fp16" as "w".

Differential Revision: D4567304

fbshipit-source-id: 87bc84fa6a45d8bfa33edb55ac1fb1cff542dbe3
2017-02-16 12:32:51 -08:00
..
docs Documenation generation to wiki 2017-02-15 16:00:44 -08:00
examples char-rnn: add comments 2017-02-10 12:20:58 -08:00
layers Rely on embedding size in split 2017-01-27 19:44:31 -08:00
mint goodbye old brewery 2017-01-04 20:58:35 -08:00
models add an option to use a resnet network instead of alexnet 2017-01-31 16:59:30 -08:00
operator_test Adding SoftmaxWithLoss operator to Shape Inference 2017-02-16 12:32:51 -08:00
_import_c_extension.py
caffe_translator.py translator fix to solve Aaron's issue 2017-02-13 11:19:13 -08:00
caffe_translator_test.py
checkpoint.py Snapshot -> Checkpoint 2017-01-27 22:29:32 -08:00
checkpoint_test.py Snapshot -> Checkpoint 2017-01-27 22:29:32 -08:00
CMakeLists.txt CMake completions work 2017-01-11 16:59:22 -08:00
cnn.py allow recurrent network gradient op to receive gradient on any combination of network output blobs 2017-02-15 16:00:45 -08: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 Better visualization for gpu training plan 2016-12-21 09:29:43 -08:00
control_test.py
convnet_benchmarks.py specify path to write htrace logs 2016-12-27 11:44:31 -08:00
convnet_benchmarks_test.py
core.py Better names for nets, steps and tasks 2017-02-09 16:33:54 -08:00
core_gradients_test.py Skip sparse tests if operators not available 2016-12-19 15:59:32 -08:00
core_test.py Shape and Type Inference Part1 2017-02-02 22:29:22 -08:00
data_parallel_model.py Use fbcollective 2017-02-07 10:48:33 -08:00
data_parallel_model_test.py lr_scale to loss_scale 2017-02-03 07:44:40 -08:00
data_workers.py Implemented logging of inputs per second 2017-02-16 12:02:05 -08:00
data_workers_test.py a couple small reliability improvements 2016-12-15 21:29:29 -08:00
dataio.py Use multi_reader directly 2017-02-06 09:59:20 -08:00
dataio_test.py
dataset.py
db_test.py
device_checker.py
dyndep.py
experiment_util.py
extension_loader.py
gradient_check_test.py Fix test cases: tensor of size 0 not supported by GPU ops yet. 2016-12-15 19:59:24 -08:00
gradient_checker.py
hsm_util.py Generate huffman tree 2017-01-19 16:14:23 -08:00
hypothesis_test.py allow recurrent network gradient op to receive gradient on any combination of network output blobs 2017-02-15 16:00:45 -08:00
hypothesis_test_util.py Allow non-batched initial recurrent states for RecurrentNetworkOp 2017-02-06 15:01:36 -08:00
introspect_vis.py Dump data for DocNN visualization 2017-02-09 13:47:07 -08:00
layer_model_helper.py get parameter blobs of a model 2017-02-15 16:00:44 -08:00
layer_model_instantiator.py Add model graph to dper_example 2017-02-07 13:03:54 -08:00
load_save_test.py Add name_overrides argument to SaveOp 2017-02-16 12:32:51 -08:00
memonger.py Fixes to topological sort, canonical blob naming, sharing final blob 2017-01-25 15:14:26 -08:00
memonger_test.py Gradient Input memory sharing using memonger blob sharing 2017-01-09 19:44:23 -08:00
mkl_test_util.py MKLDevice and MKLOperator 2016-12-15 19:59:24 -08:00
model_device_test.py Comment out NHWC Alexnet test for now 2017-01-23 13:59:29 -08:00
model_helper.py Gradient for Copy operator 2017-02-16 06:11:27 -08:00
mpi_python.cc Move mpi_python.cc to the python folder to be more consistent about source file locations. 2017-01-09 10:59:39 -08:00
muji.py
muji_test.py
net_builder.py NetBuilder: Allow to call hasattr(x, ops) out of context 2017-02-14 13:47:10 -08:00
net_builder_test.py Improvements+fixes for NetBuilder 2017-01-03 16:59:24 -08:00
net_drawer.py Add model graph to dper_example 2017-02-07 13:03:54 -08:00
net_printer.py Better names for nets, steps and tasks 2017-02-09 16:33:54 -08:00
net_printer_test.py Debug/Analysis tools for Jobs/ExecutionSteps 2017-02-06 17:31:20 -08:00
pipeline.py Better names for nets, steps and tasks 2017-02-09 16:33:54 -08:00
pybind_state.cc import_array python3 compatibility 2017-02-09 10:08:13 -08:00
pybind_state.h
pybind_state_gpu.cc
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 allow recurrent network gradient op to receive gradient on any combination of network output blobs 2017-02-15 16:00:45 -08:00
schema.py Trainer support for mobile ranking 2017-02-06 14:03:44 -08:00
schema_test.py schema.Struct.__add__ 2017-02-06 13:47:58 -08:00
scope.py
scope_test.py
session.py
session_test.py
sgd.py Migrate build_adam to python directory 2017-02-14 14:17:12 -08:00
sgd_test.py Move tests of build_sgd, build_adagrad, and build_adam to pyton directory 2017-02-14 15:32:14 -08:00
sgd_test_util.py Move tests of build_sgd, build_adagrad, and build_adam to pyton directory 2017-02-14 15:32:14 -08:00
sparse_to_dense_mask_test.py
task.py Unify train_local and train_with_distributed_readers 2017-02-09 20:46:35 -08:00
test_util.py MKL convolution operator 2017-01-23 09:59:30 -08:00
text_file_reader.py
timeout_guard.py
toy_regression_test.py
tt_core.py
tt_core_test.py
utils.py @debug decorator to make it easier to use dropin debugger 2017-01-23 09:44:26 -08:00
visualize.py
workspace.py Delete SerializeToString() call in class Model(), workspace.py 2017-02-13 23:18:03 -08:00
workspace_test.py Remove redundant and failing test of FeedBlob asserts 2016-12-22 14:59:28 -08:00