-
Notifications
You must be signed in to change notification settings - Fork 1
/
train_config.json
70 lines (63 loc) · 1.39 KB
/
train_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
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
{
"CUDA_VISIBLE_DEVICES": "1",
"max_RAM_usage": 85,
"data": {
"train_dirs": ["./DCASE_2023_Challenge_Task_7_Dataset/train.txt"],
"train_cond_dirs": null,
"test_dirs": ["./DCASE_2023_Challenge_Task_7_Dataset/eval.txt"],
"test_cond_dirs": null,
"sample_rate": 22050,
"audio_length": 88200,
"n_workers": 4
},
"model": {
"model_dir": "train_logs",
"sequential": "mamba",
"factors": [2, 2, 3, 5, 5],
"dims": [16, 32, 64, 128, 256, 512],
"bottleneck_layers": 1,
"bidirectional_bottleneck": true
},
"condition": {
"time_emb_dim": 512,
"class_emb_dim": 512,
"mid_dim": 512,
"film_type": "mamba",
"block_nums": [49, 49, 49, 49, 49, 49, 14],
"event_type": "rms",
"event_dims": {
"rms": 690,
"power": 88200,
"onset": 88200
},
"cond_prob": [0.1, 0.1]
},
"training": {
"lr": 0.0001,
"batch_size": 2,
"ema_rate": 0.999,
"scheduler_patience_epoch": 25,
"scheduler_factor": 0.8,
"scheduler_threshold": 0.001,
"n_epochs": 500
},
"logging": {
"checkpoint_id": null,
"n_epochs_to_checkpoint": 15,
"n_epochs_to_log": 15,
"n_steps_to_log": 300,
"n_steps_to_test": 300,
"n_bins": 5
},
"profiler": {
"use_profiler": false,
"wait": 298,
"warmup": 2,
"active": 3,
"repeat": 0
},
"telegram": {
"apiToken": "",
"chatID": ""
}
}