pytorch/caffe2/python/helpers
Ahmed Taei 299f293cb2 Add initializer classes to conv_nd.
Summary: Fix parameters passed to _ConvBase

Reviewed By: sunwael

Differential Revision: D5166836

fbshipit-source-id: 6c2a9fa73cf1199a5f861900554f3075a49104fc
2017-06-01 14:17:55 -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 Add initializer classes to conv_nd. 2017-06-01 14:17:55 -07:00
dropout.py New cudnn ops 2017-05-08 16:33:21 -07:00
fc.py Revert D5119830: [C2] Refactoring of the parameters step 0. Add simple tags and unify interface for params and computed_params 2017-06-01 00:02:21 -07:00
nonlinearity.py Revert D5119830: [C2] Refactoring of the parameters step 0. Add simple tags and unify interface for params and computed_params 2017-06-01 00:02:21 -07:00
normalization.py Revert D5119830: [C2] Refactoring of the parameters step 0. Add simple tags and unify interface for params and computed_params 2017-06-01 00:02:21 -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