popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

wip training

Browse Source
This commit is contained in:
Dobromir Popov
2025-12-08 22:33:36 +02:00
parent 600bee98f3
commit 5a4b0cf35b
4 changed files with 263 additions and 135 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
ANNOTATE/web/app.py
View File

@@ -2559,13 +2559,26 @@ class AnnotationDashboard:
signals = self.training_adapter.get_latest_signals()
# Get metrics from active inference sessions
# Get metrics from active inference sessions or orchestrator
metrics = {'accuracy': 0.0, 'loss': 0.0}
if hasattr(self.training_adapter, 'inference_sessions'):
for session in self.training_adapter.inference_sessions.values():
if 'metrics' in session:
metrics = session['metrics']
break
# Try to get metrics from orchestrator first (most recent)
if self.orchestrator and hasattr(self.orchestrator, 'primary_transformer_trainer'):
trainer = self.orchestrator.primary_transformer_trainer
if trainer and hasattr(trainer, 'training_history'):
history = trainer.training_history
if history.get('train_accuracy'):
metrics['accuracy'] = history['train_accuracy'][-1] if history['train_accuracy'] else 0.0
if history.get('train_loss'):
metrics['loss'] = history['train_loss'][-1] if history['train_loss'] else 0.0
# Fallback to inference session metrics
if metrics['accuracy'] == 0.0 and metrics['loss'] == 0.0:
if hasattr(self.training_adapter, 'inference_sessions'):
for session in self.training_adapter.inference_sessions.values():
if 'metrics' in session and session['metrics']:
metrics = session['metrics'].copy()
break
return jsonify({
'success': True,