fast-reid/fastreid/modeling/losses/__init__.py

9 lines
168 B
Python

# encoding: utf-8
"""
@author: l1aoxingyu
@contact: sherlockliao01@gmail.com
"""
from .cross_entroy_loss import CrossEntropyLoss
from .margin_loss import TripletLoss