caffe2.python.layers.sampling_trainable_mixin.SamplingTrainableMixin Class Reference

Inheritance diagram for caffe2.python.layers.sampling_trainable_mixin.SamplingTrainableMixin:

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 12 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 21 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 28 of file sampling_trainable_mixin.py.

---

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

• caffe2/python/layers/sampling_trainable_mixin.py