WW

Weights (parameters)

The numbers a model learns in training: the entries of its weight matrices and vectors. A model's size is counted in weights, such as 124 million for GPT-2 Small.

Everything else a model computes is derived from its input and its weights. The weights are fixed while the model runs and change only during training. The words weights and parameters mean the same thing; θ\theta stands for all of them at once.

Memory is weights times bytes per number: 124 million weights stored as 16-bit numbers take about 248 MB.

Example

WQ∈R768×64  ⟹  768⋅64=49,152 weightsW^Q \in \mathbb{R}^{768 \times 64} \;\Longrightarrow\; 768 \cdot 64 = 49{,}152 \text{ weights}

One small matrix of one attention head in GPT-2 Small.

Where the course uses it

Related terms