Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@fcakyon
fcakyon authored Dec 17, 2024
1 parent 254cce5 commit 707d667
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 707d667

Please sign in to comment.