easytexminer.losses

kd_loss

easytexminer.losses.kd_loss.soft_cross_entropy(input, targets)

easytexminer.losses.kd_loss.soft_cross_entropy_tinybert(input, targets)

easytexminer.losses.kd_loss.soft_kl_div_loss(input, targets, reduction='batchmean', **kwargs)

easytexminer.losses.kd_loss.mse_loss(inputs, targets, **kwargs)

easytexminer.losses.kd_loss.soft_input_mse_loss(inputs, targets, **kwargs)

easytexminer.losses.kd_loss.cosine_embedding_loss(input1, input2, target, **kwargs)