From 707d667ea7f160ae0113078d184d4eb37dc9a4d9 Mon Sep 17 00:00:00 2001 From: fatih akyon <34196005+fcakyon@users.noreply.github.com> Date: Tue, 17 Dec 2024 14:24:31 +0300 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 5988a05..7a9194d 100644 --- a/README.md +++ b/README.md @@ -133,7 +133,8 @@ What is the difference between this repo and vandit15's? - This repo implements loss functions as `torch.nn.Module` - In addition to class balanced losses, this repo also supports the standard versions of the cross entropy/focal loss etc. over the same API - All typos and errors in vandit15's source are fixed -- Continiously tested on PyTorch 1.13.1 and 2.5.1 +- Continuously tested on PyTorch 1.13.1 and 2.5.1 +- Automatically selects loss module device based on logits ## References