N Neurarch Architectures Models Checks Data Docs Open the app

Models / locateanything

LocateAnything-3B

Reconstructed from its own config.json with no weights read. 105K 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
3.50B
3,502,110,576 parameters
In the published checkpoint
3.83B
3,830,665,968 scalars · safetensors.total, read 2026-06-12
Delta
-8.58%

custom-code This repository ships its own modeling code (`auto_map`, e.g. `configuration_locateanything.py`), so `config.json` names a class in the repo rather than an architecture `transformers` defines. The graph below is what those config keys mean under `transformers` semantics, which is not necessarily what the repo's own file builds. A gap here is a statement about what we read, not about the checkpoint.

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
331
Will it forward-pass
Yes
Priced on
A10G (24GB)
Est. one run
$6443.58
CardMemory
T4 (16GB)weights + activationsdoes not fit
A100 (40GB)weights + activationsdoes not fit
H100 (80GB)weights + activationsfits

Structure

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

LayerTypeOutput shape
1InputInput1 × 32768
2EmbeddingEmbedding1 × 32768 × 2048
3RoPERoPE1 × 32768 × 2048
4Vision inputInput3 × 224 × 224
5PatchEmbedPatch Embed256 × 1152
6Patch_Position_EmbeddingLearned Pos Embed256 × 1152
7Vision_LN_1LayerNorm256 × 1152
8Vision_Attn_1Multi-Head Attention256 × 1152
9Vision_Add_1Add256 × 1152
10Vision_FFN_1Feed Forward256 × 1152
11Vision_LN_2LayerNorm256 × 1152
12Vision_Attn_2Multi-Head Attention256 × 1152
13Vision_Add_2Add256 × 1152
14Vision_FFN_2Feed Forward256 × 1152
15Vision_LN_3LayerNorm256 × 1152
16Vision_Attn_3Multi-Head Attention256 × 1152
17Vision_Add_3Add256 × 1152
18Vision_FFN_3Feed Forward256 × 1152
19Vision_LN_4LayerNorm256 × 1152
20Vision_Attn_4Multi-Head Attention256 × 1152
21Vision_Add_4Add256 × 1152
22Vision_FFN_4Feed Forward256 × 1152
23Vision_LN_5LayerNorm256 × 1152
24Vision_Attn_5Multi-Head Attention256 × 1152
25Vision_Add_5Add256 × 1152
26Vision_FFN_5Feed Forward256 × 1152
27Vision_LN_6LayerNorm256 × 1152
28Vision_Attn_6Multi-Head Attention256 × 1152
29Vision_Add_6Add256 × 1152
30Vision_FFN_6Feed Forward256 × 1152
31Vision_LN_7LayerNorm256 × 1152
32Vision_Attn_7Multi-Head Attention256 × 1152
33Vision_Add_7Add256 × 1152
34Vision_FFN_7Feed Forward256 × 1152
35Vision_LN_8LayerNorm256 × 1152
36Vision_Attn_8Multi-Head Attention256 × 1152
37Vision_Add_8Add256 × 1152
38Vision_FFN_8Feed Forward256 × 1152
39Vision_LN_9LayerNorm256 × 1152
40Vision_Attn_9Multi-Head Attention256 × 1152
41Vision_Add_9Add256 × 1152
42Vision_FFN_9Feed Forward256 × 1152
43Vision_LN_10LayerNorm256 × 1152
44Vision_Attn_10Multi-Head Attention256 × 1152
45Vision_Add_10Add256 × 1152
46Vision_FFN_10Feed Forward256 × 1152
47Vision_LN_11LayerNorm256 × 1152
48Vision_Attn_11Multi-Head Attention256 × 1152
49Vision_Add_11Add256 × 1152
50Vision_FFN_11Feed Forward256 × 1152
51Vision_LN_12LayerNorm256 × 1152
52Vision_Attn_12Multi-Head Attention256 × 1152
53Vision_Add_12Add256 × 1152
54Vision_FFN_12Feed Forward256 × 1152
55Vision_LN_13LayerNorm256 × 1152
56Vision_Attn_13Multi-Head Attention256 × 1152
57Vision_Add_13Add256 × 1152
58Vision_FFN_13Feed Forward256 × 1152
59Vision_LN_14LayerNorm256 × 1152
60Vision_Attn_14Multi-Head Attention256 × 1152
61Vision_Add_14Add256 × 1152
62Vision_FFN_14Feed Forward256 × 1152
63Vision_LN_15LayerNorm256 × 1152
64Vision_Attn_15Multi-Head Attention256 × 1152
65Vision_Add_15Add256 × 1152
66Vision_FFN_15Feed Forward256 × 1152
67Vision_LN_16LayerNorm256 × 1152
68Vision_Attn_16Multi-Head Attention256 × 1152
69Vision_Add_16Add256 × 1152
70Vision_FFN_16Feed Forward256 × 1152
71Vision_LN_17LayerNorm256 × 1152
72Vision_Attn_17Multi-Head Attention256 × 1152
73Vision_Add_17Add256 × 1152
74Vision_FFN_17Feed Forward256 × 1152
75Vision_LN_18LayerNorm256 × 1152
76Vision_Attn_18Multi-Head Attention256 × 1152
77Vision_Add_18Add256 × 1152
78Vision_FFN_18Feed Forward256 × 1152
79Vision_LN_19LayerNorm256 × 1152
80Vision_Attn_19Multi-Head Attention256 × 1152
81Vision_Add_19Add256 × 1152
82Vision_FFN_19Feed Forward256 × 1152
83Vision_LN_20LayerNorm256 × 1152
84Vision_Attn_20Multi-Head Attention256 × 1152
85Vision_Add_20Add256 × 1152
86Vision_FFN_20Feed Forward256 × 1152
87Vision_LN_21LayerNorm256 × 1152
88Vision_Attn_21Multi-Head Attention256 × 1152
89Vision_Add_21Add256 × 1152
90Vision_FFN_21Feed Forward256 × 1152
91Vision_LN_22LayerNorm256 × 1152
92Vision_Attn_22Multi-Head Attention256 × 1152
93Vision_Add_22Add256 × 1152
94Vision_FFN_22Feed Forward256 × 1152
95Vision_LN_23LayerNorm256 × 1152
96Vision_Attn_23Multi-Head Attention256 × 1152
97Vision_Add_23Add256 × 1152
98Vision_FFN_23Feed Forward256 × 1152
99Vision_LN_24LayerNorm256 × 1152
100Vision_Attn_24Multi-Head Attention256 × 1152
101Vision_Add_24Add256 × 1152
102Vision_FFN_24Feed Forward256 × 1152
103Vision_LN_25LayerNorm256 × 1152
104Vision_Attn_25Multi-Head Attention256 × 1152
105Vision_Add_25Add256 × 1152
106Vision_FFN_25Feed Forward256 × 1152
107Vision_LN_26LayerNorm256 × 1152
108Vision_Attn_26Multi-Head Attention256 × 1152
109Vision_Add_26Add256 × 1152
110Vision_FFN_26Feed Forward256 × 1152
111Vision_LN_27LayerNorm256 × 1152
112Vision_Attn_27Multi-Head Attention256 × 1152
113Vision_Add_27Add256 × 1152
114Vision_FFN_27Feed Forward256 × 1152
115Vision projectorProjection256 × 2048
116Vision tokensReshape1 × 256 × 2048
117Multimodal fusion (concat tokens)Concatenate1 × 33024 × 2048
118RMSNorm_1_1RMSNorm1 × 33024 × 2048
119Attention_1Grouped Query Attn1 × 33024 × 2048
120Add_1_attnAdd1 × 33024 × 2048
121RMSNorm_1_2RMSNorm1 × 33024 × 2048
122FFN_1SwiGLU1 × 33024 × 2048
123Add_1_ffnAdd1 × 33024 × 2048
124RMSNorm_2_1RMSNorm1 × 33024 × 2048
125Attention_2Grouped Query Attn1 × 33024 × 2048
126Add_2_attnAdd1 × 33024 × 2048
127RMSNorm_2_2RMSNorm1 × 33024 × 2048
128FFN_2SwiGLU1 × 33024 × 2048
129Add_2_ffnAdd1 × 33024 × 2048
130RMSNorm_3_1RMSNorm1 × 33024 × 2048
131Attention_3Grouped Query Attn1 × 33024 × 2048
132Add_3_attnAdd1 × 33024 × 2048
133RMSNorm_3_2RMSNorm1 × 33024 × 2048
134FFN_3SwiGLU1 × 33024 × 2048
135Add_3_ffnAdd1 × 33024 × 2048
136RMSNorm_4_1RMSNorm1 × 33024 × 2048
137Attention_4Grouped Query Attn1 × 33024 × 2048
138Add_4_attnAdd1 × 33024 × 2048
139RMSNorm_4_2RMSNorm1 × 33024 × 2048
140FFN_4SwiGLU1 × 33024 × 2048
141Add_4_ffnAdd1 × 33024 × 2048
142RMSNorm_5_1RMSNorm1 × 33024 × 2048
143Attention_5Grouped Query Attn1 × 33024 × 2048
144Add_5_attnAdd1 × 33024 × 2048
145RMSNorm_5_2RMSNorm1 × 33024 × 2048
146FFN_5SwiGLU1 × 33024 × 2048
147Add_5_ffnAdd1 × 33024 × 2048
148RMSNorm_6_1RMSNorm1 × 33024 × 2048
149Attention_6Grouped Query Attn1 × 33024 × 2048
150Add_6_attnAdd1 × 33024 × 2048
151RMSNorm_6_2RMSNorm1 × 33024 × 2048
152FFN_6SwiGLU1 × 33024 × 2048
153Add_6_ffnAdd1 × 33024 × 2048
154RMSNorm_7_1RMSNorm1 × 33024 × 2048
155Attention_7Grouped Query Attn1 × 33024 × 2048
156Add_7_attnAdd1 × 33024 × 2048
157RMSNorm_7_2RMSNorm1 × 33024 × 2048
158FFN_7SwiGLU1 × 33024 × 2048
159Add_7_ffnAdd1 × 33024 × 2048
160RMSNorm_8_1RMSNorm1 × 33024 × 2048
161Attention_8Grouped Query Attn1 × 33024 × 2048
162Add_8_attnAdd1 × 33024 × 2048
163RMSNorm_8_2RMSNorm1 × 33024 × 2048
164FFN_8SwiGLU1 × 33024 × 2048
165Add_8_ffnAdd1 × 33024 × 2048
166RMSNorm_9_1RMSNorm1 × 33024 × 2048
167Attention_9Grouped Query Attn1 × 33024 × 2048
168Add_9_attnAdd1 × 33024 × 2048
169RMSNorm_9_2RMSNorm1 × 33024 × 2048
170FFN_9SwiGLU1 × 33024 × 2048
171Add_9_ffnAdd1 × 33024 × 2048
172RMSNorm_10_1RMSNorm1 × 33024 × 2048
173Attention_10Grouped Query Attn1 × 33024 × 2048
174Add_10_attnAdd1 × 33024 × 2048
175RMSNorm_10_2RMSNorm1 × 33024 × 2048
176FFN_10SwiGLU1 × 33024 × 2048
177Add_10_ffnAdd1 × 33024 × 2048
178RMSNorm_11_1RMSNorm1 × 33024 × 2048
179Attention_11Grouped Query Attn1 × 33024 × 2048
180Add_11_attnAdd1 × 33024 × 2048
181RMSNorm_11_2RMSNorm1 × 33024 × 2048
182FFN_11SwiGLU1 × 33024 × 2048
183Add_11_ffnAdd1 × 33024 × 2048
184RMSNorm_12_1RMSNorm1 × 33024 × 2048
185Attention_12Grouped Query Attn1 × 33024 × 2048
186Add_12_attnAdd1 × 33024 × 2048
187RMSNorm_12_2RMSNorm1 × 33024 × 2048
188FFN_12SwiGLU1 × 33024 × 2048
189Add_12_ffnAdd1 × 33024 × 2048
190RMSNorm_13_1RMSNorm1 × 33024 × 2048
191Attention_13Grouped Query Attn1 × 33024 × 2048
192Add_13_attnAdd1 × 33024 × 2048
193RMSNorm_13_2RMSNorm1 × 33024 × 2048
194FFN_13SwiGLU1 × 33024 × 2048
195Add_13_ffnAdd1 × 33024 × 2048
196RMSNorm_14_1RMSNorm1 × 33024 × 2048
197Attention_14Grouped Query Attn1 × 33024 × 2048
198Add_14_attnAdd1 × 33024 × 2048
199RMSNorm_14_2RMSNorm1 × 33024 × 2048
200FFN_14SwiGLU1 × 33024 × 2048
201Add_14_ffnAdd1 × 33024 × 2048
202RMSNorm_15_1RMSNorm1 × 33024 × 2048
203Attention_15Grouped Query Attn1 × 33024 × 2048
204Add_15_attnAdd1 × 33024 × 2048
205RMSNorm_15_2RMSNorm1 × 33024 × 2048
206FFN_15SwiGLU1 × 33024 × 2048
207Add_15_ffnAdd1 × 33024 × 2048
208RMSNorm_16_1RMSNorm1 × 33024 × 2048
209Attention_16Grouped Query Attn1 × 33024 × 2048
210Add_16_attnAdd1 × 33024 × 2048
211RMSNorm_16_2RMSNorm1 × 33024 × 2048
212FFN_16SwiGLU1 × 33024 × 2048
213Add_16_ffnAdd1 × 33024 × 2048
214RMSNorm_17_1RMSNorm1 × 33024 × 2048
215Attention_17Grouped Query Attn1 × 33024 × 2048
216Add_17_attnAdd1 × 33024 × 2048
217RMSNorm_17_2RMSNorm1 × 33024 × 2048
218FFN_17SwiGLU1 × 33024 × 2048
219Add_17_ffnAdd1 × 33024 × 2048
220RMSNorm_18_1RMSNorm1 × 33024 × 2048
221Attention_18Grouped Query Attn1 × 33024 × 2048
222Add_18_attnAdd1 × 33024 × 2048
223RMSNorm_18_2RMSNorm1 × 33024 × 2048
224FFN_18SwiGLU1 × 33024 × 2048
225Add_18_ffnAdd1 × 33024 × 2048
226RMSNorm_19_1RMSNorm1 × 33024 × 2048
227Attention_19Grouped Query Attn1 × 33024 × 2048
228Add_19_attnAdd1 × 33024 × 2048
229RMSNorm_19_2RMSNorm1 × 33024 × 2048
230FFN_19SwiGLU1 × 33024 × 2048
231Add_19_ffnAdd1 × 33024 × 2048
232RMSNorm_20_1RMSNorm1 × 33024 × 2048
233Attention_20Grouped Query Attn1 × 33024 × 2048
234Add_20_attnAdd1 × 33024 × 2048
235RMSNorm_20_2RMSNorm1 × 33024 × 2048
236FFN_20SwiGLU1 × 33024 × 2048
237Add_20_ffnAdd1 × 33024 × 2048
238RMSNorm_21_1RMSNorm1 × 33024 × 2048
239Attention_21Grouped Query Attn1 × 33024 × 2048
240Add_21_attnAdd1 × 33024 × 2048
241RMSNorm_21_2RMSNorm1 × 33024 × 2048
242FFN_21SwiGLU1 × 33024 × 2048
243Add_21_ffnAdd1 × 33024 × 2048
244RMSNorm_22_1RMSNorm1 × 33024 × 2048
245Attention_22Grouped Query Attn1 × 33024 × 2048
246Add_22_attnAdd1 × 33024 × 2048
247RMSNorm_22_2RMSNorm1 × 33024 × 2048
248FFN_22SwiGLU1 × 33024 × 2048
249Add_22_ffnAdd1 × 33024 × 2048
250RMSNorm_23_1RMSNorm1 × 33024 × 2048
251Attention_23Grouped Query Attn1 × 33024 × 2048
252Add_23_attnAdd1 × 33024 × 2048
253RMSNorm_23_2RMSNorm1 × 33024 × 2048
254FFN_23SwiGLU1 × 33024 × 2048
255Add_23_ffnAdd1 × 33024 × 2048
256RMSNorm_24_1RMSNorm1 × 33024 × 2048
257Attention_24Grouped Query Attn1 × 33024 × 2048
258Add_24_attnAdd1 × 33024 × 2048
259RMSNorm_24_2RMSNorm1 × 33024 × 2048
260FFN_24SwiGLU1 × 33024 × 2048
261Add_24_ffnAdd1 × 33024 × 2048
262RMSNorm_25_1RMSNorm1 × 33024 × 2048
263Attention_25Grouped Query Attn1 × 33024 × 2048
264Add_25_attnAdd1 × 33024 × 2048
265RMSNorm_25_2RMSNorm1 × 33024 × 2048
266FFN_25SwiGLU1 × 33024 × 2048
267Add_25_ffnAdd1 × 33024 × 2048
268RMSNorm_26_1RMSNorm1 × 33024 × 2048
269Attention_26Grouped Query Attn1 × 33024 × 2048
270Add_26_attnAdd1 × 33024 × 2048
271RMSNorm_26_2RMSNorm1 × 33024 × 2048
272FFN_26SwiGLU1 × 33024 × 2048
273Add_26_ffnAdd1 × 33024 × 2048
274RMSNorm_27_1RMSNorm1 × 33024 × 2048
275Attention_27Grouped Query Attn1 × 33024 × 2048
276Add_27_attnAdd1 × 33024 × 2048
277RMSNorm_27_2RMSNorm1 × 33024 × 2048
278FFN_27SwiGLU1 × 33024 × 2048
279Add_27_ffnAdd1 × 33024 × 2048
280RMSNorm_28_1RMSNorm1 × 33024 × 2048
281Attention_28Grouped Query Attn1 × 33024 × 2048
282Add_28_attnAdd1 × 33024 × 2048
283RMSNorm_28_2RMSNorm1 × 33024 × 2048
284FFN_28SwiGLU1 × 33024 × 2048
285Add_28_ffnAdd1 × 33024 × 2048
286RMSNorm_29_1RMSNorm1 × 33024 × 2048
287Attention_29Grouped Query Attn1 × 33024 × 2048
288Add_29_attnAdd1 × 33024 × 2048
289RMSNorm_29_2RMSNorm1 × 33024 × 2048
290FFN_29SwiGLU1 × 33024 × 2048
291Add_29_ffnAdd1 × 33024 × 2048
292RMSNorm_30_1RMSNorm1 × 33024 × 2048
293Attention_30Grouped Query Attn1 × 33024 × 2048
294Add_30_attnAdd1 × 33024 × 2048
295RMSNorm_30_2RMSNorm1 × 33024 × 2048
296FFN_30SwiGLU1 × 33024 × 2048
297Add_30_ffnAdd1 × 33024 × 2048
298RMSNorm_31_1RMSNorm1 × 33024 × 2048
299Attention_31Grouped Query Attn1 × 33024 × 2048
300Add_31_attnAdd1 × 33024 × 2048
301RMSNorm_31_2RMSNorm1 × 33024 × 2048
302FFN_31SwiGLU1 × 33024 × 2048
303Add_31_ffnAdd1 × 33024 × 2048
304RMSNorm_32_1RMSNorm1 × 33024 × 2048
305Attention_32Grouped Query Attn1 × 33024 × 2048
306Add_32_attnAdd1 × 33024 × 2048
307RMSNorm_32_2RMSNorm1 × 33024 × 2048
308FFN_32SwiGLU1 × 33024 × 2048
309Add_32_ffnAdd1 × 33024 × 2048
310RMSNorm_33_1RMSNorm1 × 33024 × 2048
311Attention_33Grouped Query Attn1 × 33024 × 2048
312Add_33_attnAdd1 × 33024 × 2048
313RMSNorm_33_2RMSNorm1 × 33024 × 2048
314FFN_33SwiGLU1 × 33024 × 2048
315Add_33_ffnAdd1 × 33024 × 2048
316RMSNorm_34_1RMSNorm1 × 33024 × 2048
317Attention_34Grouped Query Attn1 × 33024 × 2048
318Add_34_attnAdd1 × 33024 × 2048
319RMSNorm_34_2RMSNorm1 × 33024 × 2048
320FFN_34SwiGLU1 × 33024 × 2048
321Add_34_ffnAdd1 × 33024 × 2048
322RMSNorm_35_1RMSNorm1 × 33024 × 2048
323Attention_35Grouped Query Attn1 × 33024 × 2048
324Add_35_attnAdd1 × 33024 × 2048
325RMSNorm_35_2RMSNorm1 × 33024 × 2048
326FFN_35SwiGLU1 × 33024 × 2048
327Add_35_ffnAdd1 × 33024 × 2048
328RMSNorm_36_1RMSNorm1 × 33024 × 2048
329Attention_36Grouped Query Attn1 × 33024 × 2048
330Add_36_attnAdd1 × 33024 × 2048
331RMSNorm_36_2RMSNorm1 × 33024 × 2048
332FFN_36SwiGLU1 × 33024 × 2048
333Add_36_ffnAdd1 × 33024 × 2048
334OutputOutput1 × 33024 × 2048

What the verifier says

warn"RoPE" receives input but its output is not connected. This layer will be unreachable in the forward pass. Fix: Connect the output forward, or add an Output node if this is the final layer.
dead-end
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention
infoAt 63 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 nvidia/LocateAnything-3B --plan --share