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