{ "format": "tensorviz.graph", "version": 3, "exportedAt": "2026-09-17T01:50:56.417230+00:00", "status": "static", "warningCount": 0, "document": { "schemaVersion": 2, "semantic": { "graph_id": "snapshot", "name": "SmallCNN", "revision": 0, "nodes": [ { "node_id": "input", "kind": "input", "op": "Input", "params": { "shape": "1,3,32,32" }, "in_ports": [], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "input", "name": "Input", "displayName": "Input", "description": "Model input node", "icon": "ArrowRight", "category": "io", "parameters": [ { "name": "shape", "type": "string", "defaultValue": "1,3,224,224", "required": true, "description": "Input tensor shape (comma-separated, e.g., 1,3,224,224)" }, { "name": "name", "type": "string", "defaultValue": "input", "required": false, "description": "Name for this input" } ] } } }, { "node_id": "features", "kind": "module", "op": "Module", "params": {}, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "module", "name": "Module", "displayName": "Module", "description": "Custom module group (nn.Module)", "icon": "Package", "category": "module", "parameters": [ { "name": "name", "type": "string", "defaultValue": "CustomModule", "required": true, "description": "Module class name" }, { "name": "description", "type": "string", "defaultValue": "", "required": false, "description": "Module description" } ] }, "moduleInstanceId": "features", "moduleTemplateId": "model::ImageFeatures", "isGroup": true, "moduleMembers": [ "features.conv1", "features.relu1", "features.pool1", "features.conv2", "features.relu2", "features.pool2" ], "moduleLayerCount": 6 } }, { "node_id": "flatten", "kind": "layer", "op": "Flatten", "params": {}, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "flatten", "name": "Flatten", "displayName": "Flatten", "description": "Flattens input tensor contiguously", "icon": "Maximize2", "category": "tensor", "parameters": [ { "name": "start_dim", "type": "number", "defaultValue": 1, "required": false, "description": "First dim to flatten", "min": 0 }, { "name": "end_dim", "type": "number", "defaultValue": -1, "required": false, "description": "Last dim to flatten" } ] } } }, { "node_id": "head", "kind": "layer", "op": "Linear", "params": { "in_features": 1024, "out_features": 10 }, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "linear", "name": "Linear", "displayName": "Linear", "description": "Fully Connected Layer", "icon": "Minus", "category": "linear", "parameters": [ { "name": "in_features", "type": "number", "defaultValue": 512, "required": true, "description": "Size of input features", "min": 1 }, { "name": "out_features", "type": "number", "defaultValue": 256, "required": true, "description": "Size of output features", "min": 1 }, { "name": "bias", "type": "boolean", "defaultValue": true, "required": false, "description": "If set to False, the layer will not learn an additive bias" } ] } } }, { "node_id": "output", "kind": "output", "op": "Output", "params": {}, "in_ports": [ "x" ], "out_ports": [], "metadata": { "layerType": { "id": "output", "name": "Output", "displayName": "Output", "description": "Model output node", "icon": "ArrowLeft", "category": "io", "parameters": [ { "name": "name", "type": "string", "defaultValue": "output", "required": false, "description": "Name for this output" } ] } } }, { "node_id": "features.conv1", "kind": "layer", "op": "Conv2d", "params": { "in_channels": 3, "out_channels": 8, "kernel_size": 3, "padding": 1 }, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "conv2d", "name": "Conv2d", "displayName": "Conv2D", "description": "2D Convolutional Layer", "icon": "Grid3X3", "category": "conv", "parameters": [ { "name": "in_channels", "type": "number", "defaultValue": 3, "required": true, "description": "Number of input channels", "min": 1 }, { "name": "out_channels", "type": "number", "defaultValue": 64, "required": true, "description": "Number of output channels", "min": 1 }, { "name": "kernel_size", "type": "number", "defaultValue": 3, "required": true, "description": "Size of the convolving kernel", "min": 1 }, { "name": "stride", "type": "number", "defaultValue": 1, "required": false, "description": "Stride of the convolution", "min": 1 }, { "name": "padding", "type": "number", "defaultValue": 0, "required": false, "description": "Zero-padding added to both sides", "min": 0 }, { "name": "dilation", "type": "number", "defaultValue": 1, "required": false, "description": "Spacing between kernel elements", "min": 1 }, { "name": "groups", "type": "number", "defaultValue": 1, "required": false, "description": "Number of blocked connections from input to output channels", "min": 1 }, { "name": "bias", "type": "boolean", "defaultValue": true, "required": false, "description": "If True, adds a learnable bias" } ] }, "parentId": "features" } }, { "node_id": "features.relu1", "kind": "layer", "op": "ReLU", "params": {}, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "relu", "name": "ReLU", "displayName": "ReLU", "description": "Rectified Linear Unit", "icon": "TrendingUp", "category": "activation", "parameters": [ { "name": "inplace", "type": "boolean", "defaultValue": false, "required": false, "description": "Can optionally do the operation in-place" } ] }, "parentId": "features" } }, { "node_id": "features.pool1", "kind": "layer", "op": "MaxPool2d", "params": { "kernel_size": 2 }, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "maxpool2d", "name": "MaxPool2d", "displayName": "MaxPool2D", "description": "2D Max Pooling", "icon": "Square", "category": "pool", "parameters": [ { "name": "kernel_size", "type": "number", "defaultValue": 2, "required": true, "description": "Size of the window to take a max over", "min": 1 }, { "name": "stride", "type": "number", "defaultValue": 2, "required": false, "description": "Stride of the window", "min": 1 }, { "name": "padding", "type": "number", "defaultValue": 0, "required": false, "description": "Implicit zero padding to be added", "min": 0 } ] }, "parentId": "features" } }, { "node_id": "features.conv2", "kind": "layer", "op": "Conv2d", "params": { "in_channels": 8, "out_channels": 16, "kernel_size": 3, "padding": 1 }, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "conv2d", "name": "Conv2d", "displayName": "Conv2D", "description": "2D Convolutional Layer", "icon": "Grid3X3", "category": "conv", "parameters": [ { "name": "in_channels", "type": "number", "defaultValue": 3, "required": true, "description": "Number of input channels", "min": 1 }, { "name": "out_channels", "type": "number", "defaultValue": 64, "required": true, "description": "Number of output channels", "min": 1 }, { "name": "kernel_size", "type": "number", "defaultValue": 3, "required": true, "description": "Size of the convolving kernel", "min": 1 }, { "name": "stride", "type": "number", "defaultValue": 1, "required": false, "description": "Stride of the convolution", "min": 1 }, { "name": "padding", "type": "number", "defaultValue": 0, "required": false, "description": "Zero-padding added to both sides", "min": 0 }, { "name": "dilation", "type": "number", "defaultValue": 1, "required": false, "description": "Spacing between kernel elements", "min": 1 }, { "name": "groups", "type": "number", "defaultValue": 1, "required": false, "description": "Number of blocked connections from input to output channels", "min": 1 }, { "name": "bias", "type": "boolean", "defaultValue": true, "required": false, "description": "If True, adds a learnable bias" } ] }, "parentId": "features" } }, { "node_id": "features.relu2", "kind": "layer", "op": "ReLU", "params": {}, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "relu", "name": "ReLU", "displayName": "ReLU", "description": "Rectified Linear Unit", "icon": "TrendingUp", "category": "activation", "parameters": [ { "name": "inplace", "type": "boolean", "defaultValue": false, "required": false, "description": "Can optionally do the operation in-place" } ] }, "parentId": "features" } }, { "node_id": "features.pool2", "kind": "layer", "op": "MaxPool2d", "params": { "kernel_size": 2 }, "in_ports": [ "x" ], "out_ports": [ "y" ], "metadata": { "layerType": { "id": "maxpool2d", "name": "MaxPool2d", "displayName": "MaxPool2D", "description": "2D Max Pooling", "icon": "Square", "category": "pool", "parameters": [ { "name": "kernel_size", "type": "number", "defaultValue": 2, "required": true, "description": "Size of the window to take a max over", "min": 1 }, { "name": "stride", "type": "number", "defaultValue": 2, "required": false, "description": "Stride of the window", "min": 1 }, { "name": "padding", "type": "number", "defaultValue": 0, "required": false, "description": "Implicit zero padding to be added", "min": 0 } ] }, "parentId": "features" } } ], "edges": [ { "edge_id": "edge-1", "source_node_id": "features.conv1", "target_node_id": "features.relu1", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-2", "source_node_id": "features.relu1", "target_node_id": "features.pool1", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-3", "source_node_id": "features.pool1", "target_node_id": "features.conv2", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-4", "source_node_id": "features.conv2", "target_node_id": "features.relu2", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-5", "source_node_id": "features.relu2", "target_node_id": "features.pool2", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-6", "source_node_id": "input", "target_node_id": "features.conv1", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-7", "source_node_id": "input", "target_node_id": "features", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-8", "source_node_id": "features.pool2", "target_node_id": "flatten", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-9", "source_node_id": "features", "target_node_id": "flatten", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-10", "source_node_id": "flatten", "target_node_id": "head", "source_port": "y", "target_port": "x", "metadata": {} }, { "edge_id": "edge-11", "source_node_id": "head", "target_node_id": "output", "source_port": "y", "target_port": "x", "metadata": {} } ], "inputs": [], "modules": [], "operations": [], "metadata": {} }, "view": { "graph_id": "snapshot", "node_positions": { "input": { "x": 0, "y": 160 }, "features": { "x": 340, "y": 160 }, "flatten": { "x": 680, "y": 160 }, "head": { "x": 1020, "y": 160 }, "output": { "x": 1360, "y": 160 }, "features.conv1": { "x": 24, "y": 214 }, "features.relu1": { "x": 24, "y": 364 }, "features.pool1": { "x": 24, "y": 514 }, "features.conv2": { "x": 24, "y": 664 }, "features.relu2": { "x": 24, "y": 814 }, "features.pool2": { "x": 24, "y": 964 } }, "collapsed_modules": [ "features" ], "selection": [], "layout": {} } }, "snapshotId": "snapshot-c49fae389226854a", "collaboration": { "walkthrough": { "title": "From pixels to features", "steps": [ { "id": "local-features", "title": "Look at a small neighborhood", "note": "A 3 × 3 convolution turns three color channels into eight feature maps. Padding keeps the input's 32 × 32 spatial size.", "view": { "camera": { "x": 308.5, "y": -7.5, "z": 1.25 }, "nodePositions": { "input": { "x": 0, "y": 160 }, "features": { "x": 340, "y": 160 }, "flatten": { "x": 680, "y": 160 }, "head": { "x": 1020, "y": 160 }, "output": { "x": 1360, "y": 160 }, "features.conv1": { "x": 24, "y": 214 }, "features.relu1": { "x": 24, "y": 364 }, "features.pool1": { "x": 24, "y": 514 }, "features.conv2": { "x": 24, "y": 664 }, "features.relu2": { "x": 24, "y": 814 }, "features.pool2": { "x": 24, "y": 964 } }, "expandedModuleIds": [ "features" ], "selectedNodeIds": [ "features.conv1" ], "viewport": { "width": 1100, "height": 700 } }, "anchors": [ { "target": { "kind": "node", "nodeId": "features.conv1" }, "snapshotId": "snapshot-c49fae389226854a", "exportedAt": "2026-09-17T01:50:56.417230+00:00", "modelName": "SmallCNN", "context": "{\"connections\":[\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"features.conv1\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"features.relu1\\\",\\\"targetPort\\\":\\\"x\\\"}\",\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"input\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"features.conv1\\\",\\\"targetPort\\\":\\\"x\\\"}\"],\"contents\":[],\"node\":{\"id\":\"features.conv1\",\"kind\":\"layer\",\"op\":\"Conv2d\",\"params\":{\"in_channels\":3,\"kernel_size\":3,\"out_channels\":8,\"padding\":1},\"parent\":\"features\"}}" } ] }, { "id": "downsample", "title": "Trade spatial size for features", "note": "Two 2 × 2 pooling layers reduce 32 × 32 to 16 × 16, then 8 × 8. The second convolution supplies 16 channels.", "view": { "camera": { "x": 308.5, "y": -744, "z": 1.25 }, "nodePositions": { "input": { "x": 0, "y": 160 }, "features": { "x": 340, "y": 160 }, "flatten": { "x": 680, "y": 160 }, "head": { "x": 1020, "y": 160 }, "output": { "x": 1360, "y": 160 }, "features.conv1": { "x": 24, "y": 214 }, "features.relu1": { "x": 24, "y": 364 }, "features.pool1": { "x": 24, "y": 514 }, "features.conv2": { "x": 24, "y": 664 }, "features.relu2": { "x": 24, "y": 814 }, "features.pool2": { "x": 24, "y": 964 } }, "expandedModuleIds": [ "features" ], "selectedNodeIds": [ "features.pool2" ], "viewport": { "width": 1100, "height": 700 } }, "anchors": [ { "target": { "kind": "node", "nodeId": "features.pool2" }, "snapshotId": "snapshot-c49fae389226854a", "exportedAt": "2026-09-17T01:50:56.417230+00:00", "modelName": "SmallCNN", "context": "{\"connections\":[\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"features.pool2\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"flatten\\\",\\\"targetPort\\\":\\\"x\\\"}\",\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"features.relu2\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"features.pool2\\\",\\\"targetPort\\\":\\\"x\\\"}\"],\"contents\":[],\"node\":{\"id\":\"features.pool2\",\"kind\":\"layer\",\"op\":\"MaxPool2d\",\"params\":{\"kernel_size\":2},\"parent\":\"features\"}}" } ] }, { "id": "flatten", "title": "Turn the maps into one vector", "note": "Flatten preserves the batch dimension and joins 16 × 8 × 8 values into 1,024 features per image. This is the bridge from spatial processing to classification.", "view": { "camera": { "x": -347.5, "y": 60, "z": 1.25 }, "nodePositions": { "input": { "x": 0, "y": 160 }, "features": { "x": 340, "y": 160 }, "flatten": { "x": 680, "y": 160 }, "head": { "x": 1020, "y": 160 }, "output": { "x": 1360, "y": 160 }, "features.conv1": { "x": 24, "y": 214 }, "features.relu1": { "x": 24, "y": 364 }, "features.pool1": { "x": 24, "y": 514 }, "features.conv2": { "x": 24, "y": 664 }, "features.relu2": { "x": 24, "y": 814 }, "features.pool2": { "x": 24, "y": 964 } }, "expandedModuleIds": [], "selectedNodeIds": [ "flatten" ], "viewport": { "width": 1100, "height": 700 } }, "anchors": [ { "target": { "kind": "node", "nodeId": "flatten" }, "snapshotId": "snapshot-c49fae389226854a", "exportedAt": "2026-09-17T01:50:56.417230+00:00", "modelName": "SmallCNN", "context": "{\"connections\":[\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"features\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"flatten\\\",\\\"targetPort\\\":\\\"x\\\"}\",\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"features.pool2\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"flatten\\\",\\\"targetPort\\\":\\\"x\\\"}\",\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"flatten\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"head\\\",\\\"targetPort\\\":\\\"x\\\"}\"],\"contents\":[],\"node\":{\"id\":\"flatten\",\"kind\":\"layer\",\"op\":\"Flatten\",\"params\":{}}}" } ] }, { "id": "classifier", "title": "Connect the dimensions", "note": "The linear head accepts those 1,024 features and returns 10 outputs. Changing the input image size would also change the required flattened width.", "view": { "camera": { "x": -687.5, "y": 60, "z": 1.25 }, "nodePositions": { "input": { "x": 0, "y": 160 }, "features": { "x": 340, "y": 160 }, "flatten": { "x": 680, "y": 160 }, "head": { "x": 1020, "y": 160 }, "output": { "x": 1360, "y": 160 }, "features.conv1": { "x": 24, "y": 214 }, "features.relu1": { "x": 24, "y": 364 }, "features.pool1": { "x": 24, "y": 514 }, "features.conv2": { "x": 24, "y": 664 }, "features.relu2": { "x": 24, "y": 814 }, "features.pool2": { "x": 24, "y": 964 } }, "expandedModuleIds": [], "selectedNodeIds": [ "head" ], "viewport": { "width": 1100, "height": 700 } }, "anchors": [ { "target": { "kind": "node", "nodeId": "head" }, "snapshotId": "snapshot-c49fae389226854a", "exportedAt": "2026-09-17T01:50:56.417230+00:00", "modelName": "SmallCNN", "context": "{\"connections\":[\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"flatten\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"head\\\",\\\"targetPort\\\":\\\"x\\\"}\",\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"head\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"output\\\",\\\"targetPort\\\":\\\"x\\\"}\"],\"contents\":[],\"node\":{\"id\":\"head\",\"kind\":\"layer\",\"op\":\"Linear\",\"params\":{\"in_features\":1024,\"out_features\":10}}}" } ] } ] }, "threads": [], "savedViews": [ { "id": "flatten", "title": "Turn the maps into one vector", "note": "Flatten preserves the batch dimension and joins 16 × 8 × 8 values into 1,024 features per image. This is the bridge from spatial processing to classification.", "view": { "camera": { "x": -347.5, "y": 60, "z": 1.25 }, "nodePositions": { "input": { "x": 0, "y": 160 }, "features": { "x": 340, "y": 160 }, "flatten": { "x": 680, "y": 160 }, "head": { "x": 1020, "y": 160 }, "output": { "x": 1360, "y": 160 }, "features.conv1": { "x": 24, "y": 214 }, "features.relu1": { "x": 24, "y": 364 }, "features.pool1": { "x": 24, "y": 514 }, "features.conv2": { "x": 24, "y": 664 }, "features.relu2": { "x": 24, "y": 814 }, "features.pool2": { "x": 24, "y": 964 } }, "expandedModuleIds": [], "selectedNodeIds": [ "flatten" ], "viewport": { "width": 1100, "height": 700 } }, "anchors": [ { "target": { "kind": "node", "nodeId": "flatten" }, "snapshotId": "snapshot-c49fae389226854a", "exportedAt": "2026-09-17T01:50:56.417230+00:00", "modelName": "SmallCNN", "context": "{\"connections\":[\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"features\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"flatten\\\",\\\"targetPort\\\":\\\"x\\\"}\",\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"features.pool2\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"flatten\\\",\\\"targetPort\\\":\\\"x\\\"}\",\"{\\\"kind\\\":\\\"edge\\\",\\\"source\\\":\\\"flatten\\\",\\\"sourcePort\\\":\\\"y\\\",\\\"target\\\":\\\"head\\\",\\\"targetPort\\\":\\\"x\\\"}\"],\"contents\":[],\"node\":{\"id\":\"flatten\",\"kind\":\"layer\",\"op\":\"Flatten\",\"params\":{}}}" } ] } ], "openingViewId": "flatten" } }