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


Detailed Description

Definition at line 12 of file

Member Function Documentation

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

Definition at line 21 of file

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 28 of file

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