diff options
Diffstat (limited to 'src/mlia/nn/select.py')
| -rw-r--r-- | src/mlia/nn/select.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/mlia/nn/select.py b/src/mlia/nn/select.py index 5e223fa..5a7f289 100644 --- a/src/mlia/nn/select.py +++ b/src/mlia/nn/select.py @@ -135,6 +135,7 @@ def get_optimizer( if isinstance(config, OptimizationSettings): return _get_optimizer(model, cast(OptimizationSettings, config)) + if is_list_of(config, OptimizationSettings): return _get_optimizer(model, cast(List[OptimizationSettings], config)) |