From dfc5f15ae45c484f11d21b0bd5038e31322f1da1 Mon Sep 17 00:00:00 2001 From: Adeel Hassan Date: Thu, 5 Nov 2020 12:13:29 +0500 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 5497dec..00dd386 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,7 @@ It is essentially an enhancement to cross-entropy loss and is useful for classif This repo supports importing modules through `torch.hub`. `FocalLoss` can be easily imported into your code via, for example: ```python3 focal_loss = torch.hub.load( - github='adeelh/pytorch-multi-class-focal-loss', + repo_or_dir='adeelh/pytorch-multi-class-focal-loss', model='FocalLoss', alpha=torch.tensor([.75, .25]), gamma=2, @@ -35,7 +35,7 @@ loss = focal_loss(x, y) Or: ```python3 focal_loss = torch.hub.load( - github='adeelh/pytorch-multi-class-focal-loss', + repo_or_dir='adeelh/pytorch-multi-class-focal-loss', model='focal_loss', alpha=[.75, .25], gamma=2, @@ -46,4 +46,4 @@ focal_loss = torch.hub.load( ) x, y = torch.randn(10, 2), (torch.rand(10) > .5).long() loss = focal_loss(x, y) -``` \ No newline at end of file +```