Added support for RNN
This commit is contained in:
@@ -78,6 +78,7 @@ def format_display_name(class_name):
|
||||
("MP L", "MPL"),
|
||||
("SN S", "SNS"),
|
||||
(" N A", " NA"),
|
||||
("R N N", "RNN"),
|
||||
("Adam W", "AdamW"),
|
||||
("Re L U", "ReLU"),
|
||||
("Si L U", "SiLU"),
|
||||
|
||||
Reference in New Issue
Block a user