| # PyPilot Training Script | |
| import torch | |
| from modeling_pypilot import PyPilotModel, PyPilotConfig | |
| def train_pypilot(): | |
| print("🚀 Starting PyPilot Training...") | |
| # Model configuration | |
| config = PyPilotConfig() | |
| model = PyPilotModel(config) | |
| print("✅ PyPilot model created successfully!") | |
| print(f"📊 Model parameters: {sum(p.numel() for p in model.parameters()):,}") | |
| # Training setup (simplified for now) | |
| optimizer = torch.optim.AdamW(model.parameters(), lr=0.001) | |
| print("🎯 Training ready to begin!") | |
| return model | |
| if __name__ == "__main__": | |
| train_pypilot() |