Skip to content
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.

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

CycleCL #1447

Open
4 tasks
guarin opened this issue Dec 12, 2023 · 3 comments
Open
4 tasks

CycleCL #1447

guarin opened this issue Dec 12, 2023 · 3 comments

Comments

@guarin
Copy link
Contributor

guarin commented Dec 12, 2023

CycleCL is a new contrastive learning method for cyclic videos: https://arxiv.org/abs/2311.03402

It could be an interesting addition to the package. We would have to add the following:

  •  Loading batches from a single video (ordered!)
  • Finding negative and positive examples in a batch
  • Loss -> It is basically Ntxentloss but generalized to accepting custom negative/positive pairs.
  • Augmentations
@philippmwirth
Copy link
Contributor

There already exists a triplet loss implementation in torch. The main contribution would be to implement a triplet sampler that works on multiple GPUs and then a thin wrapper to pass the triplets to the triplet loss.

@philippmwirth
Copy link
Contributor

We'd also need to add a custom projection head (they use 3D convolution).

@guarin
Copy link
Contributor Author

guarin commented Jan 11, 2024

Note: We are not planning on adding this for now.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: No status
Development

No branches or pull requests

2 participants