Added GRU support.

2025-03-25 19:25:13 -07:00
parent d9a2b827a5
commit e6abdaebdc
19 changed files with 722 additions and 14 deletions
@@ -79,6 +79,7 @@ def format_display_name(class_name):
        ("SN S", "SNS"),
        (" N A", " NA"),
        ("R N N", "RNN"),
+        ("G R U", "GRU"),
        ("Adam W", "AdamW"),
        ("Re L U", "ReLU"),
        ("Si L U", "SiLU"),