fix(optimizer): generic from_config

marcpinet · Dec 3, 2024 · 9692ece · 9692ece
1 parent d3dfa1d
commit 9692ece
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/neuralnetlib/optimizers.py b/neuralnetlib/optimizers.py
@@ -24,6 +24,10 @@ def from_config(config: dict):
             return RMSprop.from_config(config)
         elif config['name'] == 'Adam':
             return Adam.from_config(config)
+        elif config['name'] == 'AdaBelief':
+            return AdaBelief.from_config(config)
+        elif config['name'] == 'RAdam':
+            return RAdam.from_config(config)
         else:
             raise ValueError(f"Unknown optimizer name: {config['name']}")