OpenGVLab / VideoMamba

VideoMamba: State Space Model for Efficient Video Understanding
https://arxiv.org/abs/2403.06977
Apache License 2.0
660 stars 47 forks source link

Testing out VideoMamba srun configuration errors #59

Closed Tony363 closed 3 weeks ago

Tony363 commented 4 weeks ago
Screenshot 2024-05-30 122745

Hi,

I am currently trying to run the following command and would get the screen shot attached error. torchrun \ --nnodes=1 \ --nproc_per_node=1 \ --rdzv_backend=c10d \ --rdzv_endpoint=localhost:10054 \ tasks/retrieval.py \ $(dirname $0)/config.py \ output_dir $OUTPUT_DIR\ zero_shot=True \ pretrained_path=/home/tony/videomamba_m16_5M_f8_res224.pth

I would like to ask what are the correct BASE_CONFIG to run the demo?

Otherwise how do I resolve the below screen shot error?

Screenshot 2024-05-30 123048

Tony,

Andy1621 commented 4 weeks ago

Hi! It seems that you do not set the path to config.py correctly.

In mu script, the config.py and run.sh are in the same directory, thus it can runs normally.

But you run the command by tasks/retrieval.py, thus it can not find config.py in the currenet directory.

Tony363 commented 3 weeks ago

Does the exp_zx/msrvtt/run.sh support zero shot demos for my own custom data or do I have to write my own testing scripts? I am currently looking at VideoMamba/videomamba/video_mm/exp_zs/msrvtt/m16_5m/config.json and VideoMamba/videomamba/video_mm/exp_zs/msrvtt/config.py