|
Safemotion Lib
|
This is the complete list of members for classification_loss.MutiTaskCrossEntropyLoss, including all inherited members.
| __init__(self, task_key, pred_keys, gt_keys, target_tasks, weights=[1.0], data_num=None, device='cpu') | classification_loss.MutiTaskCrossEntropyLoss | |
| calc_loss(self, pred, gt, pred_key) | classification_loss.MutiTaskCrossEntropyLoss | |
| forward(self, data) | classification_loss.MutiTaskCrossEntropyLoss | |
| gt_keys | classification_loss.MutiTaskCrossEntropyLoss | |
| loss | classification_loss.MutiTaskCrossEntropyLoss | |
| pred_keys | classification_loss.MutiTaskCrossEntropyLoss | |
| target_tasks | classification_loss.MutiTaskCrossEntropyLoss | |
| task_key | classification_loss.MutiTaskCrossEntropyLoss | |
| weights | classification_loss.MutiTaskCrossEntropyLoss |