20 lines
451 B
Python
20 lines
451 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
from .cross_entropy_loss import CrossEntropyLoss
|
|
from .hard_mine_triplet_loss import TripletLoss
|
|
|
|
|
|
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 |