Searched refs:MarginCE (Results 1 – 1 of 1) sorted by relevance
62 class MarginCE(LossBase): class64 super(MarginCE, self).__init__()89 net = GradWrap(NetWithLoss(MarginCE()))