liaoxingyu 16655448c2 onnx/trt support
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00

11 lines
239 B
Python

# encoding: utf-8
"""
@author: l1aoxingyu
@contact: sherlockliao01@gmail.com
"""
from .cross_entroy_loss import CrossEntropyLoss
from .focal_loss import FocalLoss
from .triplet_loss import TripletLoss
from .circle_loss import CircleLoss