From 1d7d79fa3e8570eae310e15455ddc436f2f51bd8 Mon Sep 17 00:00:00 2001
From: Wenqi Li <wenqil@nvidia.com>
Date: Tue, 20 Apr 2021 15:49:36 +0100
Subject: [PATCH] update docstring example

Signed-off-by: Wenqi Li <wenqil@nvidia.com>
---
 monai/losses/focal_loss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/monai/losses/focal_loss.py b/monai/losses/focal_loss.py
index 5e0ccd3179..9c08658641 100644
--- a/monai/losses/focal_loss.py
+++ b/monai/losses/focal_loss.py
@@ -63,7 +63,7 @@ def __init__(
 
                 pred = torch.tensor([[1, 0], [0, 1], [1, 0]], dtype=torch.float32)
                 grnd = torch.tensor([[0], [1], [0]], dtype=torch.int64)
-                fl = FocalLoss()
+                fl = FocalLoss(to_onehot_y=True)
                 fl(pred, grnd)
 
         """