Skip to content

Commit bd6cf94

Browse files
authored
Update core.py
1 parent 1418113 commit bd6cf94

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

nam/train/core.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -487,13 +487,13 @@ def _get_lstm_config(architecture):
487487
},
488488
Architecture.FEATHER: {
489489
"num_layers": 2,
490-
"hidden_size": 8,
490+
"hidden_size": 14,
491491
"train_burn_in": 24576, #!# 4096
492492
"train_truncate": 10240, #!# 1024
493493
},
494494
Architecture.NANO: {
495-
"num_layers": 1,
496-
"hidden_size": 12,
495+
"num_layers": 2,
496+
"hidden_size": 8,
497497
"train_burn_in": 24576, #!# 4096
498498
"train_truncate": 10240, #!# 1024
499499
},

0 commit comments

Comments
 (0)