wavenet.json (1817B)
1 { 2 "_notes": [ 3 "Dev note: Ensure that tests/test_bin/test_train/test_main.py's data is ", 4 "representative of this!" 5 ], 6 "net": { 7 "name": "WaveNet", 8 "config": { 9 "layers_configs": [ 10 { 11 "condition_size": 1, 12 "input_size": 1, 13 "channels": 16, 14 "head_size": 8, 15 "kernel_size": 3, 16 "dilations": [ 17 1, 18 2, 19 4, 20 8, 21 16, 22 32, 23 64, 24 128, 25 256, 26 512 27 ], 28 "activation": "Tanh", 29 "gated": false, 30 "head_bias": false 31 }, 32 { 33 "condition_size": 1, 34 "input_size": 16, 35 "channels": 8, 36 "head_size": 1, 37 "kernel_size": 3, 38 "dilations": [ 39 1, 40 2, 41 4, 42 8, 43 16, 44 32, 45 64, 46 128, 47 256, 48 512 49 ], 50 "activation": "Tanh", 51 "gated": false, 52 "head_bias": true 53 } 54 ], 55 "head_scale": 0.02 56 } 57 }, 58 "optimizer": { 59 "lr": 0.004 60 }, 61 "lr_scheduler": { 62 "class": "ExponentialLR", 63 "kwargs": { 64 "gamma": 0.993 65 } 66 } 67 }