FM / helpers

/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/

/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
algebra.py	1296	0644	edit dl rm
arg_scope.py	1114	0644	edit dl rm
array_helpers.py	648	0644	edit dl rm
control_ops.py	625	0644	edit dl rm
conv.py	10396	0644	edit dl rm
db_input.py	381	0644	edit dl rm
dropout.py	412	0644	edit dl rm
elementwise_linear.py	1385	0644	edit dl rm
fc.py	6405	0644	edit dl rm
nonlinearity.py	1145	0644	edit dl rm
normalization.py	10891	0644	edit dl rm
pooling.py	924	0644	edit dl rm
quantization.py	261	0644	edit dl rm
tools.py	1089	0644	edit dl rm
train.py	2192	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/nonlinearity.py (1145B)

## @package nonlinearity
# Module caffe2.python.helpers.nonlinearity

from caffe2.python import core

def prelu(model, blob_in, blob_out, num_channels=1, slope_init=None,
          **kwargs):
    """PRelu"""
    slope_init = (
        slope_init if slope_init else ('ConstantFill', {'value': 0.25}))
    if model.init_params:
        slope = model.param_init_net.__getattr__(slope_init[0])(
            [],
            blob_out + '_slope',
            shape=[num_channels],
            **slope_init[1]
        )
    else:
        slope = core.ScopedBlobReference(
            blob_out + '_slope', model.param_init_net)

model.AddParameter(slope)

return model.net.PRelu([blob_in, slope], [blob_out])

def relu(model, blob_in, blob_out, use_cudnn=False, order="NCHW", **kwargs):
    """Relu."""
    if use_cudnn:
        kwargs['engine'] = 'CUDNN'
    return model.net.Relu(blob_in, blob_out, order=order, **kwargs)

def tanh(model, blob_in, blob_out, use_cudnn=False, order="NCHW", **kwargs):
    """Tanh."""
    if use_cudnn:
        kwargs['engine'] = 'CUDNN'
    return model.net.Tanh(blob_in, blob_out, order=order, **kwargs)

## @package nonlinearity # Module caffe2.python.helpers.nonlinearity from caffe2.python import core def prelu(model, blob_in, blob_out, num_channels=1, slope_init=None, **kwargs): """PRelu""" slope_init = ( slope_init if slope_init else ('ConstantFill', {'value': 0.25})) if model.init_params: slope = model.param_init_net.__getattr__(slope_init[0])( [], blob_out + '_slope', shape=[num_channels], **slope_init[1] ) else: slope = core.ScopedBlobReference( blob_out + '_slope', model.param_init_net) model.AddParameter(slope) return model.net.PRelu([blob_in, slope], [blob_out]) def relu(model, blob_in, blob_out, use_cudnn=False, order="NCHW", **kwargs): """Relu.""" if use_cudnn: kwargs['engine'] = 'CUDNN' return model.net.Relu(blob_in, blob_out, order=order, **kwargs) def tanh(model, blob_in, blob_out, use_cudnn=False, order="NCHW", **kwargs): """Tanh.""" if use_cudnn: kwargs['engine'] = 'CUDNN' return model.net.Tanh(blob_in, blob_out, order=order, **kwargs)