make_data_config.py 272 Bytes
Newer Older
yangzhong's avatar
yangzhong committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# Define dataset
dataset = dict(
    type="VideoTextDataset",
    data_path=None,
    num_frames=32,
    frame_interval=2,
    image_size=(720, 1280),
)

data_path = ''
save_path = ''
dtype = "bf16"
num_workers = 2
batch_size = 1  # now only support batch_size=1
seed = 42