Caffe2 - Python API
A deep learning, cross platform ML framework
Functions
normalization Namespace Reference

Module caffe2.python.helpers.normalization. More...

Functions

def LRN (model, blob_in, blob_out, kwargs)
 
def Softmax (model, blob_in, blob_out=None, use_cudnn=False, kwargs)
 
def InstanceNorm (model, blob_in, blob_out, dim_in, order="NCHW", kwargs)
 
def SpatialBN (model, blob_in, blob_out, dim_in, order="NCHW", kwargs)
 

Detailed Description

Module caffe2.python.helpers.normalization.

Function Documentation

◆ LRN()

def normalization.LRN (   model,
  blob_in,
  blob_out,
  kwargs 
)
LRN

Definition at line 11 of file normalization.py.

◆ Softmax()

def normalization.Softmax (   model,
  blob_in,
  blob_out = None,
  use_cudnn = False,
  kwargs 
)
Softmax.

Definition at line 21 of file normalization.py.