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

Pre-trained model weights #2

Open
nahidalam opened this issue Aug 2, 2022 · 4 comments
Open

Pre-trained model weights #2

nahidalam opened this issue Aug 2, 2022 · 4 comments

Comments

@nahidalam
Copy link

Hi

Will you publish any pre-trained model? Preferably in torchHub? I was thinking of using ViS4mer for extracting image embedding.

@md-mohaiminul
Copy link
Owner

md-mohaiminul commented Aug 3, 2022

Hi @nahidalam,
Thanks for your comment.
I will try to publish some pre-trained model weights. Do you want the pretrained model of any particular dataset? For the LVU dataset, there are 9 tasks, which tasks are you interested in?

@nahidalam
Copy link
Author

I was thinking ImageNet or any other image dataset since my goal is to get image embedding. But then I realized ViS4mer is for understanding video so I am not sure if my request makes sense.

Nonetheless wrt LVU dataset - scene/place, relationship, way of speaking - these three tasks are most relevant for me.

@md-mohaiminul
Copy link
Owner

Yes, ViS4mer is a video understanding model. However, technically you can use it for image modeling too. Anyway, I will try to release the pretrained weights for the scene/place, relationship, way of speaking tasks.

@nahidalam
Copy link
Author

Thank you for quick replay. Yes I understand it is technically possible to get embedding. I was thinking more from "meaning" perspective for my particular usecase. Maybe a model on COIN dataset makes better sense. I think a recipe on how to train ViS4mer for custom video dataset will be good so people like me can try on their own instead of taking your time. Happy to collaborate if you plan to do that sometime.

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

No branches or pull requests

2 participants