N Neurarch Architectures Models Checks Data Docs Open the app

Models / roberta

roberta-base

Reconstructed from its own config.json with no weights read. 9.7M downloads on Hugging Face.

Our count against the checkpoint

The left number comes from the graph. The right one is the number of scalars in the published weight files. Nothing on this page was tuned to make them agree.

Derived from structure
124M
124,034,304 parameters
In the published checkpoint
125M
124,697,947 scalars · safetensors.total, read 2026-09-06
Delta
-0.53%

What it costs to run

Cost is a roofline estimate on the priced GPU for 10 epochs at batch 32 over 50,000 samples (assumed; no dataset attached). GPU fit is fp32 weights plus gradients plus two Adam moments (16 bytes per parameter) with 1.3x headroom; activations are not included and grow with batch size.

Layers
50
Will it forward-pass
Yes
Priced on
A10G (24GB)
Est. one run
$1.77
CardMemory
T4 (16GB)weights + activationsfits
A100 (40GB)weights + activationsfits
H100 (80GB)weights + activationsfits

Structure

52 nodes. Output shapes are propagated from the input shape, batch dimension excluded.

LayerTypeOutput shape
1InputInput1 × 514
2EmbeddingEmbedding1 × 514 × 768
3Positional_EmbeddingLearned Pos Embed1 × 514 × 768
4Attention_1Multi-Head Attention1 × 514 × 768
5Add_1Add1 × 514 × 768
6LayerNorm_1_1LayerNorm1 × 514 × 768
7FFN_1Feed Forward1 × 514 × 768
8Attention_2Multi-Head Attention1 × 514 × 768
9Add_2Add1 × 514 × 768
10LayerNorm_2_1LayerNorm1 × 514 × 768
11FFN_2Feed Forward1 × 514 × 768
12Attention_3Multi-Head Attention1 × 514 × 768
13Add_3Add1 × 514 × 768
14LayerNorm_3_1LayerNorm1 × 514 × 768
15FFN_3Feed Forward1 × 514 × 768
16Attention_4Multi-Head Attention1 × 514 × 768
17Add_4Add1 × 514 × 768
18LayerNorm_4_1LayerNorm1 × 514 × 768
19FFN_4Feed Forward1 × 514 × 768
20Attention_5Multi-Head Attention1 × 514 × 768
21Add_5Add1 × 514 × 768
22LayerNorm_5_1LayerNorm1 × 514 × 768
23FFN_5Feed Forward1 × 514 × 768
24Attention_6Multi-Head Attention1 × 514 × 768
25Add_6Add1 × 514 × 768
26LayerNorm_6_1LayerNorm1 × 514 × 768
27FFN_6Feed Forward1 × 514 × 768
28Attention_7Multi-Head Attention1 × 514 × 768
29Add_7Add1 × 514 × 768
30LayerNorm_7_1LayerNorm1 × 514 × 768
31FFN_7Feed Forward1 × 514 × 768
32Attention_8Multi-Head Attention1 × 514 × 768
33Add_8Add1 × 514 × 768
34LayerNorm_8_1LayerNorm1 × 514 × 768
35FFN_8Feed Forward1 × 514 × 768
36Attention_9Multi-Head Attention1 × 514 × 768
37Add_9Add1 × 514 × 768
38LayerNorm_9_1LayerNorm1 × 514 × 768
39FFN_9Feed Forward1 × 514 × 768
40Attention_10Multi-Head Attention1 × 514 × 768
41Add_10Add1 × 514 × 768
42LayerNorm_10_1LayerNorm1 × 514 × 768
43FFN_10Feed Forward1 × 514 × 768
44Attention_11Multi-Head Attention1 × 514 × 768
45Add_11Add1 × 514 × 768
46LayerNorm_11_1LayerNorm1 × 514 × 768
47FFN_11Feed Forward1 × 514 × 768
48Attention_12Multi-Head Attention1 × 514 × 768
49Add_12Add1 × 514 × 768
50LayerNorm_12_1LayerNorm1 × 514 × 768
51FFN_12Feed Forward1 × 514 × 768
52OutputOutput1 × 514 × 768

What the verifier says

infoAt 12 stacked attention layers, residual-branch outputs add up; unscaled init lets activation variance grow with depth. GPT-2/LLaMA-family models scale the residual projections by depth (N(0, 0.02 / √(2L))). Fix: Scale residual output projections by depth: nn.init.normal_(w, std=0.02 / math.sqrt(2 * n_layers))
deep-attention-default-init

Do this to your own model

Same numbers, on a model in your repo, in one command. No account.

pip install neurarch-trace
neurarch-trace FacebookAI/roberta-base --plan --share