Skip to content

Commit 855668a

Browse files
committed
Update README.md.
1 parent 6830ff9 commit 855668a

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

README.md

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,18 @@ optimizer = optim.SGD(parameters, ...)
310310

311311
```
312312

313+
#### 8. Margin based classification loss
314+
Now support:
315+
1. ArcLoss
316+
2. CosLoss
317+
3. L2Softmax
318+
319+
```python
320+
from torchtoolbox.nn.loss import ArcLoss, CosLoss, L2Softmax
321+
```
322+
323+
You could use this like `nn.CrossEntropyLoss`
324+
313325
## Contribution
314326

315327
Welcome pull requests and issues!!!

0 commit comments

Comments
 (0)