# dataset settings
data_type = 'Grounding'
data_root = 'data/qvhighlights/'
data = dict(
    train=dict(
        type='RepeatDataset',
        times=4,
        dataset=dict(
            type=data_type,
            label_path=data_root + 'qvhighlights_train.jsonl',
            video_path=data_root + 'frames_224_0.5fps',
            cache_path=data_root + 'clip_b32_vid_k4',
            query_path=data_root + 'clip_b32_txt_k4',
            use_cache=True,
            min_video_len=5,
            fps=0.5,
            unit=2),
        loader=dict(batch_size=128, num_workers=4, pin_memory=True, shuffle=True)),
    val=dict(
        type=data_type,
        label_path=data_root + 'qvhighlights_val.jsonl',
        video_path=data_root + 'frames_224_0.5fps',
        cache_path=data_root + 'clip_b32_vid_k4',
        query_path=data_root + 'clip_b32_txt_k4',
        use_cache=True,
        fps=0.5,
        unit=2,
        loader=dict(batch_size=1, num_workers=4, pin_memory=True, shuffle=False)),
    test=dict(
        type=data_type,
        label_path=data_root + 'qvhighlights_test.jsonl',
        video_path=data_root + 'frames_224_0.5fps',
        cache_path=data_root + 'clip_b32_vid_k4',
        query_path=data_root + 'clip_b32_txt_k4',
        use_cache=True,
        fps=0.5,
        unit=2,
        loader=dict(batch_size=1, num_workers=4, pin_memory=True, shuffle=False)))