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

Public Member Functions

def __init__ (self, model, input_record, name='batch_distill_lr_loss', teacher_weight=0.0, filter_invalid_teacher_label=False, kwargs)
 
def add_ops (self, net)
 

Public Attributes

 threshold
 
 neg_ONE
 
 ONE
 
 output_schema
 

Detailed Description

Definition at line 18 of file batch_distill_lr_loss.py.


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