File tree Expand file tree Collapse file tree 1 file changed +8
-8
lines changed
Expand file tree Collapse file tree 1 file changed +8
-8
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments