Chore: Print model parameters

scripts/training/train.py

@@ -156,6 +156,13 @@ async def train(model, config, websocket=None, model_type="single"):
     print(f"Optimizer: {config.optimizer}")
     print(f"Batch Size: {config.batch_size}")
     print(f"Network Architecture: {config.block1}-{config.block2}-{config.block3}")
+
+    # Print model parameters
+    total_params = sum(p.numel() for p in model.parameters())
+    trainable_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
+    print(f"\nModel Parameters:")
+    print(f"Total parameters: {total_params:,}")
+    print(f"Trainable parameters: {trainable_params:,}")
     print("\nStarting training loop...")
 
     best_val_acc = 0