Jiyan Yang 043640c3eb Return top K classes ...

Reviewed By: kittipatv

Differential Revision: D5363481

fbshipit-source-id: 27ce37878434917c1a7c5f325ed77c989a1448af

2017-07-13 00:20:00 -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

Adding tanh to brew

2017-07-11 18:17:52 -07:00

layers

Return top K classes

2017-07-13 00:20:00 -07:00

mint

…

mkl

Deprecate CNNModelHelper - Inception()

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

modeling

allow param_info to set optimizer

2017-07-12 08:49:48 -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

quick fix for model_helper __init__

2017-07-12 08:49:48 -07:00

brew.py

Adding tanh to brew

2017-07-11 18:17:52 -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

…

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

Added device scope checks to data_parallel_model and data_parallel_rendevous

2017-07-12 10:47:28 -07:00

data_parallel_model.py

Added device scope checks to data_parallel_model and data_parallel_rendevous

2017-07-12 10:47:28 -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

…

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

Return top K classes

2017-07-13 00:20:00 -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

quick fix for model_helper __init__

2017-07-12 08:49:48 -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

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

Fix net_printer.py

2017-07-11 15:26:52 -07:00

optimizer_context.py

allow param_info to set optimizer

2017-07-12 08:49:48 -07:00

optimizer_test_util.py

Fp16 training initializers

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

optimizer_test.py

Add a unit-test to validate sharing learning rate between

2017-07-12 21:53:25 -07:00

optimizer.py

Optimizer: one LR op per (device, optimizer)

2017-07-12 21:17:49 -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