Skip to content

Commit 2019731

Browse files
author
Niraj Dev Pandey
authored
Update README.md
1 parent 8022751 commit 2019731

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,10 @@ The multibox loss is consist of `L1 smooth loss` and `softmax` loss. Let's see h
3939
tensor of shape (?, num_boxes, 4).
4040
y_pred: Predicted bounding boxes,
4141
tensor of shape (?, num_boxes, 4).
42+
4243
`Returns`
4344
l1_loss: L1-smooth loss, tensor of shape (?, num_boxes).
45+
4446
`References` - https://arxiv.org/abs/1504.08083
4547

4648
```python
@@ -57,8 +59,10 @@ Now let's walk through the `softmax` loss
5759
tensor of shape (?, num_boxes, num_classes).
5860
y_pred: Predicted logits,
5961
tensor of shape (?, num_boxes, num_classes).
62+
6063
`Returns`
6164
softmax_loss: Softmax loss, tensor of shape (?, num_boxes).
65+
6266

6367
```python
6468
def _softmax_loss(self, y_true, y_pred):

0 commit comments

Comments
 (0)