Interface: ModelNode

Defined in: ml/model.ts:14

One layer / op in a model graph.

Properties

flops?

optional flops?: number;

Defined in: ml/model.ts:26

Multiply-accumulates / FLOPs, if known.

---

group?

optional group?: string;

Defined in: ml/model.ts:28

Block or stage the layer belongs to ("layer1", "encoder.3", …).

---

id

id: string;

Defined in: ml/model.ts:16

Unique id — a module path (features.0), an fx node name, or a tensor name.

---

name?

optional name?: string;

Defined in: ml/model.ts:18

Display name. Defaults to id.

---

params?

optional params?: number;

Defined in: ml/model.ts:24

Trainable parameter count.

---

shape?

optional shape?: number[];

Defined in: ml/model.ts:22

Output tensor shape with the batch dim dropped, e.g. [64, 112, 112].

---

type

type: string;

Defined in: ml/model.ts:20

Layer or op class: "Conv2d", "Linear", "ReLU", "Add", …