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

Public Member Functions

def __init__ (self, model, input_record, name='batch_lr_loss', average_loss=True, jsd_weight=0.0, pos_label_target=1.0, neg_label_target=0.0, homotopy_weighting=False, log_D_trick=False, unjoined_lr_loss=False, kwargs)
 
def init_weight (self, jsd_weight, homotopy_weighting)
 
def update_weight (self, net)
 
def add_ops (self, net)
 

Public Attributes

 average_loss
 
 jsd_fuse
 
 homotopy_weighting
 
 pos_label_target
 
 neg_label_target
 
 log_D_trick
 
 unjoined_lr_loss
 
 output_schema
 
 mutex
 
 counter
 
 xent_weight
 
 jsd_weight
 

Detailed Description

Definition at line 18 of file batch_lr_loss.py.


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