feng2022's picture
add contextual_Loss
bd964f4
raw
history blame contribute delete
73 Bytes
# TODO: add supports for L1, L2 etc.
LOSS_TYPES = ['cosine', 'l1', 'l2']