fixed model training

2025-11-12 18:11:19 +02:00
parent 7cb4201bc0
commit 8354aec830
6 changed files with 9 additions and 9 deletions
--- a/NN/models/advanced_transformer_trading.py
+++ b/NN/models/advanced_transformer_trading.py
@@ -1229,7 +1229,7 @@ class TradingTransformerTrainer:
                    for k, v in batch.items()}
            
            # Use automatic mixed precision (FP16) for memory efficiency
-            with torch.cuda.amp.autocast(enabled=self.use_amp):
+            with torch.amp.autocast('cuda', enabled=self.use_amp):
                # Forward pass with multi-timeframe data
                outputs = self.model(
                    price_data_1s=batch.get('price_data_1s'),