-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.gin
51 lines (39 loc) · 1.2 KB
/
config.gin
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
#-------------DATASET--------------------
LipreadingDataset.directory = ("/home/dmitry.klimenkov/Documents/datasets/"
"lipreading_datasets/LIPS/mixed_aligned")
LipreadingDataset.labels = ("ABOUT", "PEOPLE", "ACTUALLY", "BECAUSE", "FIRST")
#-------------MAIN PARAMS--------------------
main.experiment_name = "Mobi-VSR-5W-mixed_aligned"
# directories
main.log_dir = "./logdir/"
# training
main.num_epochs = 40
main.batch_size = 38
main.lr = 1e-3
main.val_batch_size_multiplier = 2
main.weight_decay = 0
# model
main.class_weight = @get_class_weight()
# misc
main.num_workers = 4
main.check = False
main.verbose = True
#-------------MODEL PARAMS--------------------
LipNext.input_dim = 256
LipNext.nclasses = 5
LipNext.frame_len = 29
LipNext.alpha = 2
LipNext.se_block = True
LipNext.temporal_attention = False
#-------------PREPROCESS--------------------
preprocess.augmentation = True
preprocess.src_size = (112, 112)
preprocess.dst_size = (88, 88)
preprocess.video_len = 29
preprocess.augmentations_list = [
@StatefulRandomCrop(),
@StatefulRandomHorizontalFlip()
]
StatefulRandomCrop.insize = (112, 112)
StatefulRandomCrop.outsize = (88, 88)
StatefulRandomHorizontalFlip.probability = 0.5