mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
2c8b41e3f3
Summary: Adding add_weight_decay and image_input to brew module & remove `getWeights` and `getBias` from CNNModelHelper With fbgs `useWeights`, the results show that noone but add_weight_decay is using this function. I checked with oculus people, their getWeights is a different function. kennyhorror Please notice whether this is going to affect you :) Reviewed By: salexspb Differential Revision: D4945392 fbshipit-source-id: 4ef350fd81dd40a91847e9f3ebc5421eb564df32
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
## @package model_helper_api
|
|
# Module caffe2.python.model_helper_api
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import sys
|
|
import copy
|
|
|
|
# flake8: noqa
|
|
from caffe2.python.helpers.dropout import *
|
|
from caffe2.python.helpers.arg_scope import *
|
|
from caffe2.python.helpers.fc import *
|
|
from caffe2.python.helpers.pooling import *
|
|
from caffe2.python.helpers.normalization import *
|
|
from caffe2.python.helpers.nonlinearity import *
|
|
from caffe2.python.helpers.array_helpers import *
|
|
from caffe2.python.helpers.algebra import *
|
|
from caffe2.python.helpers.train import *
|
|
from caffe2.python.helpers.conv import *
|
|
from caffe2.python.helpers.tools import *
|
|
|
|
|
|
class HelperWrapper(object):
|
|
_registry = {
|
|
'arg_scope': arg_scope,
|
|
'fc': fc,
|
|
'packed_fc': packed_fc,
|
|
'fc_decomp': fc_decomp,
|
|
'fc_sparse': fc_sparse,
|
|
'fc_prune': fc_prune,
|
|
'dropout': dropout,
|
|
'max_pool': max_pool,
|
|
'average_pool': average_pool,
|
|
'lrn': lrn,
|
|
'softmax': softmax,
|
|
'instance_norm': instance_norm,
|
|
'spatial_bn': spatial_bn,
|
|
'relu': relu,
|
|
'prelu': prelu,
|
|
'concat': concat,
|
|
'depth_concat': depth_concat,
|
|
'sum': sum,
|
|
'transpose': transpose,
|
|
'iter': iter,
|
|
'accuracy': accuracy,
|
|
'conv': conv,
|
|
'conv_nd': conv_nd,
|
|
'conv_transpose': conv_transpose,
|
|
'group_conv': group_conv,
|
|
'group_conv_deprecated': group_conv_deprecated,
|
|
'image_input': image_input,
|
|
'add_weight_decay': add_weight_decay,
|
|
}
|
|
|
|
def __init__(self, wrapped):
|
|
self.wrapped = wrapped
|
|
|
|
def __getattr__(self, helper_name):
|
|
if helper_name not in self._registry:
|
|
raise AttributeError(
|
|
"Helper function {} not "
|
|
"registered.".format(helper_name)
|
|
)
|
|
|
|
def scope_wrapper(*args, **kwargs):
|
|
cur_scope = get_current_scope()
|
|
new_kwargs = copy.deepcopy(cur_scope.get(helper_name, {}))
|
|
new_kwargs.update(kwargs)
|
|
return self._registry[helper_name](*args, **new_kwargs)
|
|
|
|
scope_wrapper.__name__ = helper_name
|
|
return scope_wrapper
|
|
|
|
def Register(self, helper):
|
|
name = helper.__name__
|
|
if name in self._registry:
|
|
raise AttributeError(
|
|
"Helper {} already exists. Please change your "
|
|
"helper name.".format(name)
|
|
)
|
|
self._registry[name] = helper
|
|
|
|
def has_helper(self, helper_or_helper_name):
|
|
helper_name = (
|
|
helper_or_helper_name
|
|
if isinstance(helper_or_helper_name, basestring) else
|
|
helper_or_helper_name.__name__
|
|
)
|
|
return helper_name in self._registry
|
|
|
|
|
|
sys.modules[__name__] = HelperWrapper(sys.modules[__name__])
|