VideoSearch-R1 Charades-STA Stage 2

This is the Stage 2 VideoSearch-R1 checkpoint trained for Charades-STA, presented in the paper VideoSearch-R1: Iterative Video Retrieval and Reasoning via Soft Query Refinement.

Usage

Use with the VideoSearch-R1 codebase:

bash scripts/data_construct/download_preextracted_data.bash charades
EVAL_GPUS=0 bash scripts/inference/inference.bash charades --checkpoint VideoSearchR1/charades-stage2
Downloads last month
99
Safetensors
Model size
2B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for VideoSearchR1/charades-stage2

Finetuned
(333)
this model

Space using VideoSearchR1/charades-stage2 1

Paper for VideoSearchR1/charades-stage2