fix: judge loss is not exists

dataelement · Oct 28, 2024 · e6bf853 · e6bf853
1 parent 5128e06
commit e6bf853
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backend/bisheng/api/services/finetune.py b/src/backend/bisheng/api/services/finetune.py
@@ -406,7 +406,7 @@ def parse_log_data(cls, log_data: str) -> List[Dict[str, str]]:
             sub_data = {'step': None, 'loss': None}
             elem = elem.strip()
             elem_data = json.loads(elem)
-            if elem_data['loss'] is None:
+            if elem_data.get('loss', None) is None:
                 continue
             sub_data['step'] = elem_data['current_steps']
             sub_data['loss'] = elem_data['loss']