-
Notifications
You must be signed in to change notification settings - Fork 1
/
example_config.json
39 lines (27 loc) · 1.06 KB
/
example_config.json
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
{
"help": "Standard Parameters for training , use this as template.",
"data_augmentation": {
"video_image_param": { "input_size" : 256 , "output_size": 224},
"video_augmentation": {"RandomHorizontalFlip": true, "scales": [1, 0.875, 0.75, 0.66], "ColorJitter": false, "RandomGrayscale": 0.0, "GaussianBlur":false},
"audio_image_param": {"window_sizes": [25, 50, 100], "hop_sizes": [10, 25, 50], "n_mels": 224, "eps": 1e-6, "spec_size": [3, 224, 224], "num_segments": 1, "m_segments": 1},
"audio_augmentation": {"status":true, "random_shift_waveform": [1.0, 1.0]}
},
"model": {
"video_segments": 4,
"audio_segments": 0,
"arch": "resnet50_timm",
"pretrain": "imagenet",
"dropout": 0.5
},
"shift_temporal": {
"status": true,
"f_div": 4,
"shift_depth": 4,
"n_insert": 2,
"m_insert": 0
},
"training_param": {"epochs": 50, "batch_size": 3, "num_workers": 40},
"optimizer_param": {"lr": 5.0e-2, "lr_decay": [0.1, 5, 0.01, 25, 0.001,45], "momentum": 0.9, "gd": 20 ,"weight_decay": 1e-4},
"save_epoch": [90],
"results_folder": "results"
}