Skip to content

Commit 1418113

Browse files
authored
Update core.py
added more #!# to find more of my changes.
1 parent 75a3e32 commit 1418113

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

nam/train/core.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -476,26 +476,26 @@ def _get_lstm_config(architecture):
476476
Architecture.STANDARD: {
477477
"num_layers": 2,
478478
"hidden_size": 47,
479-
"train_burn_in": 24576,
480-
"train_truncate": 10240,
479+
"train_burn_in": 24576, #!# 4096
480+
"train_truncate": 10240, #!# 1024
481481
},
482482
Architecture.LITE: {
483483
"num_layers": 2,
484484
"hidden_size": 21,
485-
"train_burn_in": 24576,
486-
"train_truncate": 10240,
485+
"train_burn_in": 24576, #!# 4096
486+
"train_truncate": 10240, #!# 1024
487487
},
488488
Architecture.FEATHER: {
489489
"num_layers": 2,
490490
"hidden_size": 8,
491-
"train_burn_in": 24576,
492-
"train_truncate": 10240,
491+
"train_burn_in": 24576, #!# 4096
492+
"train_truncate": 10240, #!# 1024
493493
},
494494
Architecture.NANO: {
495495
"num_layers": 1,
496496
"hidden_size": 12,
497-
"train_burn_in": 24576,
498-
"train_truncate": 10240,
497+
"train_burn_in": 24576, #!# 4096
498+
"train_truncate": 10240, #!# 1024
499499
},
500500
}[architecture]
501501

0 commit comments

Comments
 (0)