GithubHelp home page GithubHelp logo

salesforce / alpro Goto Github PK

View Code? Open in Web Editor NEW
183.0 7.0 18.0 317 KB

Align and Prompt: Video-and-Language Pre-training with Entity Prompts

License: BSD 3-Clause "New" or "Revised" License

Shell 0.90% Python 99.10%
vision-and-language video-language video-text-retrieval video-question-answering representation-learning prompt-learning

alpro's People

Contributors

dxli94 avatar svc-scm avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar

alpro's Issues

An academic issues on your paper

In the video encoder part, the output is {v_cls, v_1, ..., v_k} (so the dimension is (k+1)*d)
therefore, the dimension of multi-modal video-text encoder is (k+N_t+1)*d
but according to paper: you claim that the dimension of multi-modal video-text encoder is (N_v+N_t+1)*d
I'm confused about this...

MSR-VTT Zero-shot

Hi ,

I saw "We pre-train ALPRO for 100k iterations, roughly equivalent to 10 epochs" in your paper. So there will be ten checkpoints, which one is the zero-shot checkpoint for testing MSRVTT, and how to choose?

Looking forward to your reply, thanks!

Using multiple GPUs vs single GPU

Hi,

Congratulations on the amazing work. Will there be any difference in performance if I use just a single GPU and what are the changes to be made in eg: msvd_qa.json?

Thank you.

Zero-shot for MSRVTT retrieval results

Hi, I have tried to use the provided pretrained checkpoint of ALPRO to finetune on MSRVTT.

Before start training, it would test on the validation set, and it gives me the results:

{'text2video': {'r1': 17.9, 'r5': 40.0, 'r10': 50.9, 'medianR': 10.0, 'meanR': 57.259}, 'video2text': {'r1': 16.0, 'r5': 34.6, 'r10': 46.0, 'medianR': 13.0, 'meanR': 61.448}

I think it does not match the results presented in the paper which achieves R@1 with 24.1, R@5 with 44.7. I am wondering why it would happen?

installing environment

Thanks for the sharing code! I was trying to set up the environment but met with some problem especially on installing apex; I wonder if it is possible to provide a .yaml file that can used to create the environment using only Conda? or a docker container for setting up the environment? Thanks!

Inference

Hi,

In the inference we always load the best model. However, after fine-tuning there is no checkpoint named $OUTPUT_DIR/ckpt/model_step_best.pt. Can you point to the line in the code where the best checkpoint is saved?

Thank you.

Test on one GPU

Hello,

Thank you very much for open-sourcing this code. I would like to try running it locally, but I only have one GPU, and I've encountered several issues while trying to install the Horovod library, making it impossible for me to proceed. Could you please let me know if there is a version for single GPU testing that doesn't require Horovod installation? Alternatively, could you provide instructions for setting up the environment directly through Conda?

Looking forward to your reply, Very thanks!

Weight Decay

Hi, as stated in the issue, the ALPRO does use weight decay. But I did not find the process that passing the parameter "weight_decay" during the optimizer initialization.

optimizer = OptimCls(model.parameters(), lr=opts.learning_rate, betas=opts.betas)

Issues loading CC3M

Hi, I am trying to pretrain the model using CC3M dataset,

After the first iteration (batch), the program would stuck, and give the following warning.
horovod/common/stall_inspector.cc:105] One or more tensors were submitted to be reduced, gathered or broadcasted by subset of ranks and are waiting for remainder of ranks for more than 60 seconds. This may indicate that different ranks are trying to submit different tensors or that only subset of ranks is submitting tensors, which will cause deadlock.

Is there any way that I can avoid this?

MSR-VTT dataset split

Hi,
Thanks for sharing the code!

I saw "use 7k videos for training and report results on the 1k test split" in your paper. When I downloaded the MSR-VTT dataset, there are only 7K train sets and 3K test sets, but no val dataset. Could you share the code for dividing the dataset to avoid discrepancies in results?

Looking forward to your reply.

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.