Esempio n. 1
0
File: loss.py Progetto: yyaqi/Dragon
 def LayerSetup(self, bottom):
     loss = _ops.SmoothL1Loss(bottom, **self.arguments)
     if self._loss_weight is not None: loss *= self._loss_weight
     return loss
Esempio n. 2
0
 def Setup(self, bottom):
     super(SmoothL1LossLayer, self).Setup(bottom)
     loss = ops.SmoothL1Loss(bottom, **self._param)
     if self._loss_weight is not None: loss *= self._loss_weight
     return loss
Esempio n. 3
0
 def Setup(self, bottom):
     super(SmoothL1LossLayer, self).Setup(bottom)
     return ops.SmoothL1Loss(bottom, **self._param)