diff --git a/neuralnetlib/optimizers.py b/neuralnetlib/optimizers.py index a941c83..59bc4a5 100644 --- 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']}")