tweaks, try live trading
This commit is contained in:
@ -1489,7 +1489,20 @@ class EnhancedRealtimeTrainingSystem:
|
||||
|
||||
outputs = model(features_tensor)
|
||||
|
||||
loss = criterion(outputs, targets_tensor)
|
||||
# Extract logits from model output (model returns a dictionary)
|
||||
if isinstance(outputs, dict):
|
||||
logits = outputs['logits']
|
||||
elif isinstance(outputs, tuple):
|
||||
logits = outputs[0] # First element is usually logits
|
||||
else:
|
||||
logits = outputs
|
||||
|
||||
# Ensure logits is a tensor
|
||||
if not isinstance(logits, torch.Tensor):
|
||||
logger.error(f"CNN output is not a tensor: {type(logits)}")
|
||||
return 0.0
|
||||
|
||||
loss = criterion(logits, targets_tensor)
|
||||
|
||||
loss.backward()
|
||||
optimizer.step()
|
||||
|
Reference in New Issue
Block a user