pytorch/caffe2/python/helpers
Alexander Sidorov 016f72537a ModelHelper.create_param, Initializer abstraction and ParameterInfo for optimizers
Summary:
This is going to unblock Nvidia in their work on adding fp16
support to Caffe2. I discussed this with kennyhorror before to make
sure this fits into his work on parameter sharing.

Reviewed By: kennyhorror

Differential Revision: D5127797

fbshipit-source-id: 4db155d320b1862570c23b77c4252bdacbf2296f
2017-05-25 22:03:15 -07:00
..
algebra.py rename model_helpers to brew and lowercase all helper functions 2017-04-24 15:52:26 -07:00
arg_scope.py arg_scope for model_helper 2017-04-24 15:52:25 -07:00
array_helpers.py rename model_helpers to brew and lowercase all helper functions 2017-04-24 15:52:26 -07:00
conv.py rename model_helpers to brew and lowercase all helper functions 2017-04-24 15:52:26 -07:00
dropout.py New cudnn ops 2017-05-08 16:33:21 -07:00
fc.py ModelHelper.create_param, Initializer abstraction and ParameterInfo for optimizers 2017-05-25 22:03:15 -07:00
nonlinearity.py rename model_helpers to brew and lowercase all helper functions 2017-04-24 15:52:26 -07:00
normalization.py Fix LRN schema for cuDNN op 2017-05-22 08:33:34 -07:00
pooling.py New cudnn ops 2017-05-08 16:33:21 -07:00
tools.py Adding video data layer for caffe2 2017-05-05 14:16:38 -07:00
train.py Adding add_weight_decay and image_input to brew module 2017-04-25 16:03:58 -07:00