-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathconfigs.py.example
30 lines (28 loc) · 1003 Bytes
/
configs.py.example
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# Path to the downloaded CLIP official weights.
# See: https://github.com/openai/CLIP/blob/a9b1bf5920416aaeaec965c25dd9e8f98c864f16/clip/clip.py#L30
CLIP_VIT_B16_PATH = ''
CLIP_VIT_L14_PATH = ''
# Whether cuDNN should be temporarily disable for 3D depthwise convolution.
# For some PyTorch builds the built-in 3D depthwise convolution may be much
# faster than the cuDNN implementation. You may experiment with your specific
# environment to find out the optimal option.
DWCONV3D_DISABLE_CUDNN = True
# Configuration of datasets. The required fields are listed for Something-something-v2 (ssv2)
# and Kinetics-400 (k400). Fill in the values to use the datasets, or add new datasets following
# these examples.
DATASETS = {
'ssv2': dict(
TRAIN_ROOT='',
VAL_ROOT='',
TRAIN_LIST='',
VAL_LIST='',
NUM_CLASSES=174,
),
'k400': dict(
TRAIN_ROOT='',
VAL_ROOT='',
TRAIN_LIST='',
VAL_LIST='',
NUM_CLASSES=400,
),
}