Skip to content

Add self-supervised losses #1695

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ahmedtaha100 wants to merge 1 commit into
base: main
Choose a base branch
from
+1,073 −0
Open

Add self-supervised losses #1695

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions docs/api/losses.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,16 @@ Losses
.. autosummary::
:toctree: generated/

barlow_twins_loss
binary_dice_loss
byol_loss
convex_kl_divergence
cosine_distance
cosine_similarity
ctc_loss
ctc_loss_with_forward_probs
dice_loss
dino_loss
generalized_kl_divergence
hinge_loss
huber_loss
Expand All @@ -32,6 +35,7 @@ Losses
safe_softmax_cross_entropy
sigmoid_binary_cross_entropy
sigmoid_focal_loss
simsiam_loss
smooth_labels
softmax_cross_entropy
softmax_cross_entropy_with_integer_labels
Expand Down
4 changes: 4 additions & 0 deletions optax/losses/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,10 @@
from optax.losses._segmentation import binary_dice_loss
from optax.losses._segmentation import dice_loss
from optax.losses._segmentation import multiclass_generalized_dice_loss
from optax.losses._self_supervised import barlow_twins_loss
from optax.losses._self_supervised import byol_loss
from optax.losses._self_supervised import dino_loss
from optax.losses._self_supervised import ntxent
from optax.losses._self_supervised import simsiam_loss
from optax.losses._self_supervised import triplet_margin_loss
from optax.losses._smoothing import smooth_labels
Loading
Loading