pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

Aapo Kyrola 192e0546bf fix for back-and-forth models, pass reference instead of copy

Summary:
akirillov again presented me with a memonger-bug: his model that has kind of a 'back-and-forth structure' where blobs are passed left and right in a ladder-like structure, revealed a bug in memonger: I should pass the set of free blobs as a reference, not a copy so that the recyclings are properly accounted for. Hard to explain.

Since we have the graph verifier, we can be more confident with these changes.

I also added some helpful debug to the graph verifier.

Differential Revision: D5396925

fbshipit-source-id: 0bffb3a0bf8532afcd6b5bc9331c779768a8c5c5

2017-07-11 10:52:14 -07:00

docs

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

examples

Fixed typo

2017-06-23 14:02:40 -07:00

helpers

Conv frontend: checking engine and use_cudnn

2017-06-27 09:47:48 -07:00

layers

Simplifying Random Fourier Features and layer test

2017-07-11 00:40:53 -07:00

mint

…

mkl

Deprecate CNNModelHelper - Inception()

2017-06-15 14:03:27 -07:00

modeling

Skip fp16 initializer test for CPU-only builds

2017-06-19 12:21:25 -07:00

models

fast simple-net memonger for C++

2017-07-06 15:17:07 -07:00

operator_test

Implemented GRUCell

2017-07-10 17:52:25 -07:00

predictor

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

rnn

Moved sigmoid, tanh, and _prepare_lstm (renamed) to a util file.

2017-07-10 17:52:22 -07:00

_import_c_extension.py

…

attention.py

Unrolled test for AttentionCell

2017-06-25 17:21:24 -07:00

brew_test.py

Fixing broken Python tests

2017-06-08 13:34:46 -07:00

brew.py

Core unit test fixes for Python 3

2017-06-23 13:22:16 -07:00

caffe_translator_test.py

…

caffe_translator.py

Read pretrained weights using binary mode in caffe_translator.py

2017-07-08 10:17:57 -07:00

checkpoint_test.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

checkpoint.py

…

CMakeLists.txt

…

cnn.py

cnnmodelhelper deprecate warning

2017-05-18 23:35:26 -07:00

context_test.py

…

context.py

…

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

add debug information when there is blob version mismatch

2017-06-30 16:22:46 -07:00

core_test.py

single trainer hybrid device

2017-06-27 22:06:30 -07:00

core.py

Fix communication_schema decoding

2017-07-02 13:04:20 -07:00

crf.py

Deprecate CNNModelHelper in python/crf.py

2017-06-14 08:49:27 -07:00

data_parallel_model_test.py

Add synchronization barrier API to data parallel model

2017-07-06 09:21:19 -07:00

data_parallel_model.py

fix a bug in the report function of Data_Parallel

2017-07-07 13:03:46 -07:00

data_workers_test.py

fix a rare race condition by initializing scratch blobs beforehand

2017-06-26 10:18:18 -07:00

data_workers.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

dataio_test.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

dataio.py

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -07:00

dataset.py

Add random shuffle through the data to the benchmark workflow

2017-06-16 13:22:46 -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

…

empty.so

Adding video data layer for caffe2

2017-05-05 14:16:38 -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

…

gradient_check_test.py

Cos, Sin, and Abs operators

2017-07-03 22:18:32 -07:00

gradient_checker.py

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -07:00

gru_cell.py

Implemented GRUCell

2017-07-10 17:52:25 -07:00

hsm_util.py

…

hypothesis_test_util.py

Add min_satisfying_examples

2017-06-29 12:48:01 -07:00

hypothesis_test.py

Cos, Sin, and Abs operators

2017-07-03 22:18:32 -07:00

layer_model_helper.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

layer_model_instantiator.py

Remove map() and filter() in favor of comprehensions

2017-05-30 15:32:58 -07:00

layer_test_util.py

Core unit test fixes for Python 3

2017-06-23 13:22:16 -07:00

layers_test.py

Simplifying Random Fourier Features and layer test

2017-07-11 00:40:53 -07:00

load_save_test.py

…

lstm_benchmark.py

Added flags to lstm, convnet and sparse_nn_benchmarks to print out operators

2017-06-30 23:47:04 -07:00

memonger_test.py

fast simple-net memonger for C++

2017-07-06 15:17:07 -07:00

memonger.py

fix for back-and-forth models, pass reference instead of copy

2017-07-11 10:52:14 -07:00

mkl_test_util.py

…

model_device_test.py

Deprecate CNNModelHelper in caffe2/python/model_device_test.py

2017-06-22 15:37:17 -07:00

model_helper.py

make GetComputedParams work like GetParams

2017-07-10 12:30:44 -07:00

mpi_python.cc

Fix pybind11 module name for MPI helpers

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

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

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

net_builder.py

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -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

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

optimizer_test_util.py

Fp16 training initializers

2017-06-01 08:34:46 -07:00

optimizer_test.py

add_weight_decay + restore weight decay to resnet50_trainer

2017-06-02 14:16:56 -07:00

optimizer.py

Adding dedup aggregator options to sgd optimizer

2017-07-04 02:10:18 -07:00

parallelize_gpu_bmuf_distributed_test.py

Add distributed BMUF implementation.

2017-06-21 16:18:11 -07:00

pipeline.py

Enable runtime cloning of tasks.

2017-06-21 03:18:20 -07:00

predictor_constants.py

…

pybind_state_gpu.cc

…

pybind_state_mkl.cc

…

pybind_state.cc

fast simple-net memonger for C++

2017-07-06 15:17:07 -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

…

record_queue.py

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -07:00

recurrent.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

rnn_cell.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

schema_test.py

Add __sub__ function for schema.Struct

2017-06-28 11:24:01 -07:00

schema.py

IndexHash

2017-07-07 23:06:11 -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

Allow tasks/execution_steps to be cloned at runtime

2017-06-20 22:32:07 -07:00

sparse_to_dense_mask_test.py

String-related fixes for Python 3

2017-05-26 16:04:32 -07:00

task.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00

test_util.py

…

text_file_reader.py

…

timeout_guard.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -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

Fast path for serializing large floating-point tensors to protobuf

2017-07-10 17:52:22 -07:00

visualize.py

Python 3 compatible integer division

2017-07-06 11:47:12 -07:00

workspace_test.py

Core unit test fixes for Python 3

2017-06-23 13:22:16 -07:00

workspace.py

Dict fixes/improvements and unittest targets for Python 3 in caffe2 core

2017-06-29 17:05:41 -07:00