Added LSTM support

2025-03-26 02:43:05 -07:00
parent e6abdaebdc
commit fce22ba1dd
14 changed files with 721 additions and 5 deletions
@@ -80,6 +80,7 @@ def format_display_name(class_name):
        (" N A", " NA"),
        ("R N N", "RNN"),
        ("G R U", "GRU"),
+        ("L S T M", "LSTM"),
        ("Adam W", "AdamW"),
        ("Re L U", "ReLU"),
        ("Si L U", "SiLU"),