Skip to content

Commit 2dafb35

Browse files
authored
Update wavenet.json
Lowered lr. Increased gamma. Changed dilations... etc
1 parent 199797a commit 2dafb35

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

nam_full_configs/models/wavenet.json

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313
"channels": 16,
1414
"head_size": 8,
1515
"kernel_size": 3,
16-
"dilations": [1,2,4,8,16,32,64,128,256,512],
17-
"activation": "Tanh",
16+
"dilations": [512,384,192,96,32,16,8,4,2,1,24],
17+
"activation": "Hardtanh",
1818
"gated": false,
1919
"head_bias": false
2020
},
@@ -24,8 +24,8 @@
2424
"channels": 8,
2525
"head_size": 1,
2626
"kernel_size": 3,
27-
"dilations": [1,2,4,8,16,32,64,128,256,512],
28-
"activation": "Tanh",
27+
"dilations": [512,384,192,96,32,16,8,4,2,1,24],
28+
"activation": "Hardtanh",
2929
"gated": false,
3030
"head_bias": true
3131
}
@@ -34,12 +34,12 @@
3434
}
3535
},
3636
"optimizer": {
37-
"lr": 0.004
37+
"lr": 0.002
3838
},
3939
"lr_scheduler": {
4040
"class": "ExponentialLR",
4141
"kwargs": {
42-
"gamma": 0.997
42+
"gamma": 0.9985
4343
}
4444
}
4545
}

0 commit comments

Comments
 (0)