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

Public Member Functions

def __init__ (self, model, input_record, num_to_collect, name='last_n_window_collector', kwargs)
 
def add_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)
 

Public Attributes

 num_to_collect
 
 last_n
 
 next_blob
 
 mutex
 
 num_visited_blob
 
 output_schema
 
- Public Attributes inherited from caffe2.python.layers.layers.ModelLayer
 name
 
 model
 
 kwargs
 
 request_only
 
 eval_output_schema
 
 tags
 
 params
 

Detailed Description

Collect last-N samples from input record. If you have complex data,
use PackRecords to pack it before using this layer.

This layer is not thread safe.

Definition at line 27 of file last_n_window_collector.py.


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