| 12345678910111213141516171819202122232425262728293031323334353637 |
- # @package regularizer_context
- # Module caffe2.python.normalizer_context
- from caffe2.python import context
- from caffe2.python.modifier_context import (
- ModifierContext, UseModifierBase)
- class NormalizerContext(ModifierContext, context.DefaultManaged):
- """
- provide context to allow param_info to have different normalizers
- """
- def has_normalizer(self, name):
- return self._has_modifier(name)
- def get_normalizer(self, name):
- assert self.has_normalizer(name), (
- "{} normalizer is not provided!".format(name))
- return self._get_modifier(name)
- class UseNormalizer(UseModifierBase):
- '''
- context class to allow setting the current context.
- Example usage with layer:
- normalizers = {'norm1': norm1, 'norm2': norm2}
- with UseNormalizer(normalizers):
- norm = NormalizerContext.current().get_normalizer('norm1')
- layer(norm=norm)
- '''
- def _context_class(self):
- return NormalizerContext
|
