ETRI IVCL 1.0.0
Acceleration SW Platform for Ondevice
Public Member Functions | Public Attributes | List of all members
total_loss_layer.TotalLossLayer Class Reference
Inheritance diagram for total_loss_layer.TotalLossLayer:

Public Member Functions

def setup (self, bottom, top)
 
def reshape (self, bottom, top)
 
def absolute_difference (self, labels, predictions, weights=1.0, reduction='mean')
 
def setup (self, bottom, top)
 
def reshape (self, bottom, top)
 
def absolute_difference (self, labels, predictions, weights=1.0, reduction='mean')
 

Public Attributes

 src
 
 gt
 
 loss
 

Member Function Documentation

◆ absolute_difference() [1/2]

def total_loss_layer.TotalLossLayer.absolute_difference (   self,
  labels,
  predictions,
  weights = 1.0,
  reduction = 'mean' 
)

◆ absolute_difference() [2/2]

def total_loss_layer.TotalLossLayer.absolute_difference (   self,
  labels,
  predictions,
  weights = 1.0,
  reduction = 'mean' 
)

◆ reshape() [1/2]

def total_loss_layer.TotalLossLayer.reshape (   self,
  bottom,
  top 
)

◆ reshape() [2/2]

def total_loss_layer.TotalLossLayer.reshape (   self,
  bottom,
  top 
)

◆ setup() [1/2]

def total_loss_layer.TotalLossLayer.setup (   self,
  bottom,
  top 
)

◆ setup() [2/2]

def total_loss_layer.TotalLossLayer.setup (   self,
  bottom,
  top 
)

Member Data Documentation

◆ gt

total_loss_layer.TotalLossLayer.gt

◆ loss

total_loss_layer.TotalLossLayer.loss

◆ src

total_loss_layer.TotalLossLayer.src

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