YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

Spark Mini-GPT

A small GPT model I built from scratch in PyTorch, trained on a short repeated story about a robot named Spark.

Architecture

  • 12-layer transformer
  • 768-dim embeddings, 12 attention heads
  • GPT-2 tokenizer (50,257 vocab)

Training

  • ~7,600 tokens (a short story repeated 50 times)
  • 10 epochs with AdamW
  • Final loss: ~0.007

Why I made this

This was a class project to learn how GPT models actually work

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support