Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
caffe2.python.layers.fc.FC Class Reference
Inheritance diagram for caffe2.python.layers.fc.FC:
caffe2.python.layers.sampling_trainable_mixin.SamplingTrainableMixin caffe2.python.layers.layers.ModelLayer

Public Member Functions

def __init__ (self, model, input_record, output_dims, weight_init=None, bias_init=None, weight_optim=None, bias_optim=None, name='fc', weight_reg=None, bias_reg=None, clip_param=None, max_fc_size=None, axis=1, kwargs)
 
def param_blobs (self)
 
- Public Member Functions inherited from caffe2.python.layers.sampling_trainable_mixin.SamplingTrainableMixin
def __init__ (self, args, kwargs)
 
def param_blobs (self)
 
def train_param_blobs (self)
 
def train_param_blobs (self, blobs)
 
def add_ops (self, net)
 
def add_train_ops (self, net)
 
- Public Member Functions inherited from caffe2.python.layers.layers.ModelLayer
def __init__ (self, model, prefix, input_record, predict_input_record_fields=None, tags=None, kwargs)
 
def get_type (self)
 
def predict_input_record (self)
 
def input_record (self)
 
def predict_output_schema (self)
 
def predict_output_schema (self, output_schema)
 
def output_schema (self)
 
def output_schema (self, output_schema)
 
def get_parameters (self)
 
def get_fp16_compatible_parameters (self)
 
def get_memory_usage (self)
 
def add_init_params (self, init_net)
 
def create_param (self, param_name, shape, initializer, optimizer, ps_param=None, regularizer=None)
 
def get_next_blob_reference (self, name)
 
def add_operators (self, net, init_net=None, context=InstantiationContext.TRAINING)
 
def add_ops (self, net)
 
def add_eval_ops (self, net)
 
def add_train_ops (self, net)
 
def add_ops_to_accumulate_pred (self, net)
 
def add_param_copy_operators (self, net)
 
def export_output_for_metrics (self)
 
def export_params_for_metrics (self)
 

Static Public Member Functions

def calculate_fc_output_dims (max_fc_size, input_dim, output_dim)
 

Public Attributes

 axis
 
 clip_args
 
 output_dim_vec
 
 w
 
 b
 
 w_vec
 
 b_vec
 
 output_schema
 
- Public Attributes inherited from caffe2.python.layers.sampling_trainable_mixin.SamplingTrainableMixin
 train_param_blobs
 
- Public Attributes inherited from caffe2.python.layers.layers.ModelLayer
 name
 
 model
 
 kwargs
 
 request_only
 
 precomputation_request_only
 
 precomputation_object_only
 
 eval_output_schema
 
 tags
 
 params
 

Detailed Description

Definition at line 15 of file fc.py.


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