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

runtime fix

Browse Source
This commit is contained in:
Dobromir Popov
2025-11-22 15:32:42 +02:00
parent 2f28fcc89a
commit 8b784412b6
2 changed files with 47 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
ANNOTATE/web/app.py
View File

@@ -631,24 +631,33 @@ class AnnotationDashboard:
self.training_adapter.orchestrator = self.orchestrator
logger.info("TradingOrchestrator initialized")
# Load the specific model
# Check if the specific model is already initialized
if model_name == 'Transformer':
logger.info("Loading Transformer model...")
self.orchestrator.load_transformer_model()
self.loaded_models['Transformer'] = self.orchestrator.primary_transformer_trainer
logger.info("Transformer model loaded successfully")
logger.info("Checking Transformer model...")
if self.orchestrator.primary_transformer:
self.loaded_models['Transformer'] = self.orchestrator.primary_transformer
logger.info("Transformer model loaded successfully")
else:
logger.warning("Transformer model not initialized in orchestrator")
return
elif model_name == 'CNN':
logger.info("Loading CNN model...")
self.orchestrator.load_cnn_model()
self.loaded_models['CNN'] = self.orchestrator.cnn_model
logger.info("CNN model loaded successfully")
logger.info("Checking CNN model...")
if self.orchestrator.cnn_model:
self.loaded_models['CNN'] = self.orchestrator.cnn_model
logger.info("CNN model loaded successfully")
else:
logger.warning("CNN model not initialized in orchestrator")
return
elif model_name == 'DQN':
logger.info("Loading DQN model...")
self.orchestrator.load_dqn_model()
self.loaded_models['DQN'] = self.orchestrator.dqn_agent
logger.info("DQN model loaded successfully")
logger.info("Checking DQN model...")
if self.orchestrator.rl_agent:
self.loaded_models['DQN'] = self.orchestrator.rl_agent
logger.info("DQN model loaded successfully")
else:
logger.warning("DQN model not initialized in orchestrator")
return
else:
logger.warning(f"Unknown model name: {model_name}")