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 | |
Definition at line 12 of file sampling_trainable_mixin.py.
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.