File size: 623 Bytes
86f5eb2 02220b5 86f5eb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# 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() |