往下拉回到首頁
一張顯卡就能練出 AI?RTX 5080 從零訓練 2.35 億參數 LLM,這樣買卡才划算

一張顯卡就能練出 AI?RTX 5080 從零訓練 2.35 億參數 LLM,這樣買卡才划算

235M param LLM from scratch on a single RTX 5080

Hey everyone, Been working on this for a while and figured I'd share it here too. I made a small transformer language model completely from scratch in PyTorch. No pretrained weights, no HuggingFace downloads. Every parameter was trained from raw text on a single consumer GPU. Current release is Plasma 1.0 (235M params, 18 layers, hidden size 1024). LLaMA-style: GQA with 16 query heads and 4 KV heads (head_dim 64), SwiGLU FFN with 2816 intermediate dim, RoPE with theta 10000, RMSNorm pre-norm,