diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index 9dc28c8..5ee8dde 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -129,7 +129,7 @@ def train() -> None: { 'source_path': CONFIG.get('training.model', 'source_path'), 'target_path': CONFIG.get('training.model', 'target_path'), - 'learning_rate': CONFIG.getfloat('training.trainer', 'learning_rate'), + 'learning_rate': CONFIG.getfloat('training.trainer', 'learning_rate') } config_common =\ {