demonet.json (1291B)
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 "input_size": 1, 12 "condition_size": 1, 13 "channels": 8, 14 "head_size": 4, 15 "kernel_size": 3, 16 "dilations": [1, 2, 4, 8, 16, 32, 64], 17 "activation": "Tanh", 18 "gated": false, 19 "head_bias": false 20 }, 21 { 22 "condition_size": 1, 23 "input_size": 8, 24 "channels": 4, 25 "head_size": 1, 26 "kernel_size": 3, 27 "dilations": [128, 256, 512, 1, 2, 4, 8, 16, 32, 64, 128, 256, 512], 28 "activation": "Tanh", 29 "gated": false, 30 "head_bias": true 31 } 32 ], 33 "head_scale": 0.02 34 } 35 }, 36 "optimizer": { 37 "lr": 0.004 38 }, 39 "lr_scheduler": { 40 "class": "ExponentialLR", 41 "kwargs": { 42 "gamma": 0.993 43 } 44 } 45 }