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

Public Member Functions

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 Attributes

 train_param_blobs
 

Detailed Description

Definition at line 27 of file sampling_trainable_mixin.py.

Member Function Documentation

def caffe2.python.layers.sampling_trainable_mixin.SamplingTrainableMixin.param_blobs (   self)
List of parameter blobs for prediction net

Definition at line 36 of file sampling_trainable_mixin.py.

def caffe2.python.layers.sampling_trainable_mixin.SamplingTrainableMixin.train_param_blobs (   self)
If train_param_blobs is not set before used, default to param_blobs

Definition at line 43 of file sampling_trainable_mixin.py.


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