We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8022751 commit 2019731Copy full SHA for 2019731
README.md
@@ -39,8 +39,10 @@ The multibox loss is consist of `L1 smooth loss` and `softmax` loss. Let's see h
39
tensor of shape (?, num_boxes, 4).
40
y_pred: Predicted bounding boxes,
41
42
+
43
`Returns`
44
l1_loss: L1-smooth loss, tensor of shape (?, num_boxes).
45
46
`References` - https://arxiv.org/abs/1504.08083
47
48
```python
@@ -57,8 +59,10 @@ Now let's walk through the `softmax` loss
57
59
tensor of shape (?, num_boxes, num_classes).
58
60
y_pred: Predicted logits,
61
62
63
64
softmax_loss: Softmax loss, tensor of shape (?, num_boxes).
65
66
67
68
def _softmax_loss(self, y_true, y_pred):
0 commit comments