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
7fa4acab9b
pytorch/caffe2/python
History
Bor-Yiing Su 7fa4acab9b Loads only the model blobs from the checkpoints. 
Summary:
To evaluate from checkpoints, we need to load a model from the checkpoints.
However, the checkpoints store way more blobs than the blobs needed by the
model. This function enables the model builder to load only the blobs
associated with the model to the workspace. After that, the model builder
can evaluate the model from the populated workspace.

Reviewed By: azzolini

Differential Revision: D4751414

fbshipit-source-id: a7a420228d681fc2dcfd8573cf69a97b1abc2ef3
2017-03-27 10:02:11 -07:00
..
docs Documenation generation to wiki 2017-02-15 16:00:44 -08:00
examples Remove more instances of batch_size 2017-03-19 22:31:30 -07:00
layers allows to specify output names for functional layers 2017-03-23 13:18:58 -07:00
mint goodbye old brewery 2017-01-04 20:58:35 -08:00
models exec_net --> predict_net 2017-03-23 16:31:49 -07:00
operator_test Registering GPU version of PackSegments using GPUFallbackOp 2017-03-24 16:01:53 -07:00
_import_c_extension.py fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
attention.py Removing unnecessary Copy after fixing gradients for external parameters 2017-03-23 15:04:22 -07:00
caffe_translator_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
caffe_translator.py translator fix 2017-03-22 15:09:13 -07:00
checkpoint_test.py Loads only the model blobs from the checkpoints. 2017-03-27 10:02:11 -07:00
checkpoint.py Loads only the model blobs from the checkpoints. 2017-03-27 10:02:11 -07:00
CMakeLists.txt CMake completions work 2017-01-11 16:59:22 -08:00
cnn.py Add ConvNd to model helper 2017-03-22 15:47:48 -07:00
context_test.py Make ContextManager thread-safe 2017-02-13 19:45:35 -08:00
context.py Make ContextManager thread-safe 2017-02-13 19:45:35 -08:00
control_test.py fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
control.py Better visualization for gpu training plan 2016-12-21 09:29:43 -08:00
convnet_benchmarks_test.py chunky sync - build scripts to be written 2016-07-21 10:16:42 -07:00
convnet_benchmarks.py Convnet benchmark cudnn_ws 2017-03-02 15:32:37 -08:00
core_gradients_test.py add inference for gradient ops + a couple of missing shape inference functions + fix to scalars 2017-02-28 23:33:32 -08:00
core_test.py NextScopedBlob with well-defined behavior and respect namescope 2017-02-16 17:16:36 -08:00
core.py fixes to make data parallel model work for RecurrentNet + test case 2017-03-14 15:48:07 -07:00
crf.py CRF layer in caffe2 2017-03-23 22:02:02 -07:00
data_parallel_model_test.py fixes to make data parallel model work for RecurrentNet + test case 2017-03-14 15:48:07 -07:00
data_parallel_model.py add gpu support for caffe2-seq2seq 2017-03-17 05:19:14 -07:00
data_workers_test.py close blobs queues when stopping + test 2017-02-27 10:07:57 -08:00
data_workers.py Remove use of logging module and np.random.randint() due to deadlocks with forks 2017-03-01 03:32:56 -08: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 Stop multi_reader if we run out of data before max_examples 2017-03-10 18:03:57 -08:00
dataset.py fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
db_test.py Fix db_test under tsan 2016-11-29 15:18:37 -08:00
device_checker.py chunky sync 2016-09-06 15:55:19 -07:00
dyndep.py fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
experiment_util.py fix external_loggers 2017-03-16 13:47:58 -07:00
extension_loader.py fbsync 2016-10-07 13:08:53 -07:00
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 fbsync 2016-10-07 13:08:53 -07:00
hsm_util.py Generate huffman tree 2017-01-19 16:14:23 -08:00
hypothesis_test_util.py Allow use of ReversePackedSegs operator in CUDA context 2017-03-09 15:03:55 -08:00
hypothesis_test.py add AccumulateHistogramOp 2017-03-08 19:37:32 -08:00
introspect_vis.py User input (Conv out, etc.) 2017-03-08 13:49:45 -08:00
layer_model_helper.py Skip optimizer when param doesn't have gradient and optimizer is not set 2017-03-24 22:18:34 -07:00
layer_model_instantiator.py Skip optimizer when param doesn't have gradient and optimizer is not set 2017-03-24 22:18:34 -07:00
layers_test.py allows to specify output names for functional layers 2017-03-23 13:18:58 -07:00
load_save_test.py Improve error message from LogFileDB on missing file 2017-03-10 23:31:28 -08:00
lstm_benchmark.py LSTM benchmark (Caffe2 RNN based) 2017-02-28 23:17:26 -08:00
memonger_test.py Gradient Input memory sharing using memonger blob sharing 2017-01-09 19:44:23 -08:00
memonger.py Fixes to topological sort, canonical blob naming, sharing final blob 2017-01-25 15:14:26 -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 add gpu support for caffe2-seq2seq 2017-03-17 05:19:14 -07: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_test.py chunky sync - build scripts to be written 2016-07-21 10:16:42 -07:00
muji.py fbsync 2016-10-07 13:08:53 -07:00
net_builder_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
net_builder.py Improve "reporter net" design 2017-02-21 20:17:40 -08:00
net_drawer.py Add model graph to dper_example 2017-02-07 13:03:54 -08:00
net_printer_test.py Debug/Analysis tools for Jobs/ExecutionSteps 2017-02-06 17:31:20 -08:00
net_printer.py Add task outputs and stop signals to net_printer 2017-03-07 01:21:40 -08:00
optimizer_test_util.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
optimizer_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
optimizer.py refactor and modulize optimizers 2017-03-07 18:46:47 -08:00
pipeline.py Better names for nets, steps and tasks 2017-02-09 16:33:54 -08:00
pybind_state_gpu.cc Cudnn v6 2017-02-28 17:46:33 -08:00
pybind_state_mkl.cc Expose MKLMemory to the Python Feed and Fetch interface, and misc changes 2016-11-29 15:18:36 -08:00
pybind_state.cc Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00
pybind_state.h bugfix for Windows, esp. VS 2017 2017-03-21 05:17:59 -07:00
python_op_test.py Allow PythonOp to access the workspace 2016-12-05 11:53:26 -08:00
queue_util.py More perf stats for BlobsQueue 2017-03-24 14:03:28 -07:00
record_queue.py chunky sync 2016-09-06 15:55:19 -07:00
recurrent.py Don't do copy for param_grad in backward_step_net 2017-03-24 02:22:33 -07:00
schema_test.py Struct nested field name lookup supports List 2017-03-24 18:17:19 -07:00
schema.py Struct nested field name lookup supports List 2017-03-24 18:17:19 -07:00
scope_test.py fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
scope.py fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
session_test.py NextScopedBlob with well-defined behavior and respect namescope 2017-02-16 17:16:36 -08:00
session.py Default LocalSession to current workspace. 2017-03-01 16:03:18 -08:00
sparse_to_dense_mask_test.py Fix few more operators to handle empty batches correctly. 2016-11-29 15:18:37 -08:00
task.py Gather perf counters for distributed jobs 2017-02-21 22:06:25 -08:00
test_util.py MKL convolution operator 2017-01-23 09:59:30 -08:00
text_file_reader.py fix typo in TextFileReader 2017-02-21 14:02:48 -08:00
timeout_guard.py Euthanize a process with timeout 2017-03-01 11:38:11 -08:00
toy_regression_test.py sync 2016-08-10 11:02:15 -07:00
tt_core_test.py sync 2016-08-10 11:02:15 -07:00
tt_core.py sync 2016-08-10 11:02:15 -07:00
utils.py Extract net and blobs assignment to separate functions 2017-03-24 13:17:24 -07:00
visualize.py chunky sync 2016-05-13 14:43:48 -07:00
workspace_test.py Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00
workspace.py Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00