pytorch/caffe2/python/helpers
Alexander Sidorov a3b8c459d4 Revamp MNIST tutorial
Summary:
Main changes:

1. Move reader creation to Brew in order to be consistent and avoid a wild use of param_init_net
2. Use optimizers for training function, avoid manual optimizer construction
3. Add MLP mode (a default)
4. Fix a bunch of too verbose comments and add a bit of new explanations
Closes https://github.com/caffe2/caffe2/pull/1760

Differential Revision: D6749059

Pulled By: salexspb

fbshipit-source-id: 9dfbbb2d9772a74a0300c2e404a92e791f7cc593
2018-01-26 09:17:31 -08:00
..
algebra.py Re-license to Apache 2017-09-28 16:22:00 -07:00
arg_scope.py Re-license to Apache 2017-09-28 16:22:00 -07:00
array_helpers.py Re-license to Apache 2017-09-28 16:22:00 -07:00
control_ops.py Add if and while ops to brew 2017-12-05 17:33:34 -08:00
conv.py Re-license to Apache 2017-09-28 16:22:00 -07:00
db_input.py Revamp MNIST tutorial 2018-01-26 09:17:31 -08:00
dropout.py Disable cudnn dropout 2017-10-04 17:24:09 -07:00
elementwise_linear.py Re-license to Apache 2017-09-28 16:22:00 -07:00
fc.py Added FP16 compute support to FC Op 2017-10-30 17:03:51 -07:00
nonlinearity.py Re-license to Apache 2017-09-28 16:22:00 -07:00
normalization.py update transformer code for layer_norm() API change 2017-12-01 20:19:31 -08:00
pooling.py Re-license to Apache 2017-09-28 16:22:00 -07:00
tools.py Add missing is_test argument in ImageInput ops 2017-10-10 10:03:13 -07:00
train.py Re-license to Apache 2017-09-28 16:22:00 -07:00