forked from PaddlePaddle/PaddleVideo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
bmn.yaml
100 lines (90 loc) · 2.61 KB
/
bmn.yaml
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
MODEL: #MODEL field
framework: "BMNLocalizer"
backbone:
name: "BMN"
tscale: 100
dscale: 100
prop_boundary_ratio: 0.5
num_sample: 32
num_sample_perbin: 3
loss:
name: "BMNLoss"
tscale: 100
dscale: 100
DATASET: #DATASET field
batch_size: 4 #single card bacth size
test_batch_size: 1
num_workers: 8
train:
format: "BMNDataset"
file_path: "data/dataset/bmn_data/activitynet_1.3_annotations.json"
subset: "train"
valid:
format: "BMNDataset"
file_path: "data/dataset/bmn_data/activitynet_1.3_annotations.json"
subset: "validation"
test:
format: "BMNDataset"
test_mode: True
file_path: "data/dataset/bmn_data/activitynet_1.3_annotations.json"
subset: "validation"
PIPELINE: #PIPELINE field
train: #Mandotary, indicate the pipeline to deal with the training data, associate to the 'paddlevideo/loader/pipelines/'
load_feat:
name: "LoadFeat"
feat_path: "data/dataset/bmn_data/fix_feat_100"
transform: #Mandotary, image transfrom operator
- GetMatchMap:
tscale: 100
- GetVideoLabel:
tscale: 100
dscale: 100
valid: #Mandotary, indicate the pipeline to deal with the training data, associate to the 'paddlevideo/loader/pipelines/'
load_feat:
name: "LoadFeat"
feat_path: "data/dataset/bmn_data/fix_feat_100"
transform: #Mandotary, image transfrom operator
- GetMatchMap:
tscale: 100
- GetVideoLabel:
tscale: 100
dscale: 100
test: #Mandatory, indicate the pipeline to deal with the validing data. associate to the 'paddlevideo/loader/pipelines/'
load_feat:
name: "LoadFeat"
feat_path: "data/dataset/bmn_data/fix_feat_100"
transform: #Mandotary, image transfrom operator
- GetMatchMap:
tscale: 100
- GetVideoLabel:
tscale: 100
dscale: 100
OPTIMIZER: #OPTIMIZER field
name: 'Adam'
learning_rate:
iter_step: True
name: 'CustomPiecewiseDecay'
boundaries: [4200]
values: [0.001, 0.0001]
weight_decay:
name: 'L2'
value: 1e-4
METRIC:
name: 'BMNMetric'
tscale: 100
dscale: 100
file_path: "data/dataset/bmn_data/activitynet_1.3_annotations.json"
ground_truth_filename: "data/dataset/bmn_data/activity_net_1_3_new.json"
subset: "validation"
output_path: "data/bmn/BMN_Test_output"
result_path: "data/bmn/BMN_Test_results"
INFERENCE:
name: 'BMN_Inference_helper'
feat_dim: 400
dscale: 100
tscale: 100
result_path: "data/bmn/BMN_INFERENCE_results"
model_name: BMN
epochs: 9 #Mandatory, total epoch
log_level: "INFO"
resume_from: "" #checkpoint path.