Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
caffe2.python.modeling.initializers.Initializer Class Reference
Inheritance diagram for caffe2.python.modeling.initializers.Initializer:
caffe2.python.modeling.initializers.PseudoFP16Initializer caffe2.python.modeling.initializers.ReversePseudoFP16Initializer

Public Member Functions

def __init__ (self, operator_name=None, kwargs)
 
def update (self, operator_name, kwargs)
 
def create_param (self, param_name, init_net, shape)
 

Public Attributes

 operator_name
 
 operator_kwargs
 

Detailed Description

This class abstracts out parameter creation. One can come up with a new
Initializer in order to implement more complex parameter initializaion logic

Definition at line 12 of file initializers.py.


The documentation for this class was generated from the following file: