From 094f19483e890721d20b9197b6ca23f13da791ff Mon Sep 17 00:00:00 2001
From: fcakyon <34196005+fcakyon@users.noreply.github.com>
Date: Thu, 21 Jul 2022 23:59:04 +0300
Subject: [PATCH] update readme

---
 README.md | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/README.md b/README.md
index 7cfb151..671896b 100644
--- a/README.md
+++ b/README.md
@@ -44,11 +44,15 @@ labels = torch.tensor([0]) # 1 batch
 # focal loss
 focal_loss = Loss(loss_type="focal_loss")
 loss = focal_loss(logits, labels)
+```
 
+```python
 # cross-entropy loss
 ce_loss = Loss(loss_type="cross_entropy")
 loss = ce_loss(logits, labels)
+```
 
+```python
 # binary cross-entropy loss
 bce_loss = Loss(loss_type="binary_cross_entropy")
 loss = bce_loss(logits, labels)
@@ -74,7 +78,9 @@ focal_loss = Loss(
     class_balanced=True
 )
 loss = focal_loss(logits, labels)
+```
 
+```python
 # class-balanced cross-entropy loss
 ce_loss = Loss(
     loss_type="cross_entropy",
@@ -82,7 +88,9 @@ ce_loss = Loss(
     class_balanced=True
 )
 loss = ce_loss(logits, labels)
+```
 
+```python
 # class-balanced binary cross-entropy loss
 bce_loss = Loss(
     loss_type="binary_cross_entropy",