THU-IAR's picture
Upload 198 files
2d06dcc verified
from torch import nn
from .KCL import KCL
from .MCL import MCL
from .SupConLoss import SupConLoss
loss_map = {
'CrossEntropyLoss': nn.CrossEntropyLoss(),
'KCL': KCL(),
'MCL': MCL(),
'SupConLoss': SupConLoss()
}