22 lines
526 B
Python
22 lines
526 B
Python
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
from .cross_entropy_loss import CrossEntropyLabelSmooth
|
||
|
|
from .hard_mine_triplet_loss import TripletLoss
|
||
|
|
from .center_loss import CenterLoss
|
||
|
|
from .ring_loss import RingLoss
|
||
|
|
|
||
|
|
|
||
|
|
def DeepSupervision(criterion, xs, y):
|
||
|
|
"""
|
||
|
|
Args:
|
||
|
|
- criterion: loss function
|
||
|
|
- xs: tuple of inputs
|
||
|
|
- y: ground truth
|
||
|
|
"""
|
||
|
|
loss = 0.
|
||
|
|
for x in xs:
|
||
|
|
loss += criterion(x, y)
|
||
|
|
loss /= len(xs)
|
||
|
|
return loss
|