Net Mesh Scope
English · Русский
English
Net Mesh Scope is a Visual Studio Code extension for exploring neural-network structures stored in PyTorch checkpoint files.
It opens .pt, .pth, .ckpt, and .bin files, discovers model, ema, and state_dict branches, and presents modules, parameters, buffers, and available connections in an interactive interface.
What the extension is for
Net Mesh Scope helps you:
- understand an unfamiliar model quickly;
- inspect a third-party checkpoint;
- examine layers, parameters, tensor shapes, and data types;
- compare two revisions of a model;
- prepare architecture diagrams for documentation;
- read explanations and equations for common neural-network operations.
Main features
- Interactive architecture view for the whole network or a selected module.
- Compact, left-to-right, and top-to-bottom layouts.
- Zooming, mouse panning, block folding, and minimap navigation.
- Four synchronized views: Architecture, Modules, Tensors, and Parameters.
- Search by name, path, type, and tensor shape.
- Filters for modules, parameters, buffers, and regular tensors.
- Inspection of channels, kernel size, stride, padding, groups, data types, and other saved settings.
- Built-in offline guides with KaTeX equations, small diagrams, and practical explanations.
- Automatic official PyTorch documentation for a single-block view when an internet connection is available.
- Incoming and outgoing connection highlighting for the selected layer.
- Checkpoint comparison by structure, layer settings, and tensor shapes and types.
- Output-shape, MAC, and arithmetic-FLOP estimates for
Conv2d and Linear.
- SVG and PNG export for the diagram or selected block.
- JSON export of inspected metadata.
- Light, dark, and high-contrast VS Code theme support.
Checkpoint handling
The bundled reader processes ZIP and pickle structures as data. Referenced objects are not invoked; the extension reconstructs the available tensor metadata, saved attributes, and module hierarchy.
Supported data
- Modern ZIP-based PyTorch 1.6+ checkpoints.
.pt, .pth, .ckpt, and .bin files containing a PyTorch checkpoint or state dictionary.nn.Module models, EMA copies, and plain state dictionaries.- Saved Ultralytics routes.
- Recognized
Sequential container order.
Current limitations
- Legacy pre-1.6 PyTorch checkpoints are not supported yet.
- ONNX is not supported and is not interpreted as a PyTorch checkpoint.
- Arbitrary
forward logic cannot be fully reconstructed from weights alone. The viewer shows module containment when routing information is unavailable.
- Plain state dictionaries may not preserve layer types or the exact role of every tensor.
- Comparison checks inspected metadata and structure, not numerical weight values.
- FLOP estimates are not runtime benchmarks or measurements of actual memory use.
Usage
- Right-click a
.pt, .pth, .ckpt, or .bin file in the VS Code Explorer.
- Select Open Model Structure.
- You can also run the command from the Command Palette and choose a file afterward.
- Use the tabs to inspect architecture, modules, tensors, and parameters.
- Select a layer to open its properties and built-in guide.
Click Help in the viewer toolbar to open the complete guide for controls, formulas, comparison, export, and operation estimates.
Русский
Net Mesh Scope — расширение Visual Studio Code для исследования структуры нейронных сетей, сохранённых в PyTorch checkpoint-файлах.
Оно открывает .pt, .pth, .ckpt и .bin, находит ветви model, ema и state_dict, а затем показывает модули, параметры, буферы и доступные связи в интерактивном интерфейсе.
Для чего нужно расширение
Net Mesh Scope помогает:
- быстро понять устройство незнакомой модели;
- изучить сторонний checkpoint;
- посмотреть слои, параметры, формы и типы тензоров;
- сравнить две версии модели;
- подготовить схему архитектуры для документации;
- получить справку и формулы для распространённых операций нейросети.
Основные возможности
- Интерактивная схема всей модели или выбранного модуля.
- Компактная, горизонтальная и вертикальная раскладки.
- Масштабирование, перемещение мышью, сворачивание блоков и мини-карта.
- Четыре синхронизированных представления: Architecture, Modules, Tensors и Parameters.
- Поиск по именам, путям, типам и формам тензоров.
- Фильтры модулей, параметров, буферов и обычных тензоров.
- Инспектор каналов, kernel size, stride, padding, групп, типов данных и других сохранённых настроек.
- Встроенная офлайн-справка с формулами KaTeX, мини-графиками и пояснениями.
- Автоматическое отображение официальной документации PyTorch для одиночного блока при доступном интернете.
- Подсветка входящих и исходящих связей выбранного слоя.
- Сравнение checkpoint по структуре, настройкам слоёв, формам и типам тензоров.
- Оценка формы выхода, MACs и арифметических FLOPs для
Conv2d и Linear.
- Экспорт схемы или выбранного блока в SVG и PNG.
- Экспорт исследованных метаданных в JSON.
- Поддержка светлой, тёмной и high-contrast тем VS Code.
Обработка checkpoint
Встроенный reader обрабатывает структуры ZIP и pickle как данные. Объекты, на которые ссылается файл, не вызываются: расширение восстанавливает доступные метаданные тензоров, сохранённые атрибуты и дерево модулей.
Поддерживаемые данные
- Современные ZIP-based checkpoint PyTorch 1.6 и новее.
- Файлы
.pt, .pth, .ckpt и .bin, содержащие PyTorch checkpoint или state dictionary.
- Модели
nn.Module, EMA-копии и обычные state dictionaries.
- Сохранённые маршруты Ultralytics.
- Последовательные связи контейнеров
Sequential.
Ограничения
- Legacy checkpoint PyTorch до версии 1.6 пока не поддерживаются.
- ONNX пока не поддерживается и не интерпретируется как PyTorch checkpoint.
- Произвольную логику
forward невозможно полностью восстановить только по весам. Если маршруты не сохранены, отображается вложенность модулей.
- Для обычного state dictionary типы слоёв и роли некоторых тензоров могут быть неизвестны.
- Сравнение проверяет метаданные и структуру, но не числовые значения весов.
- Оценка FLOPs не является измерением времени работы или фактического потребления памяти.
Использование
- Нажмите правой кнопкой на
.pt, .pth, .ckpt или .bin в проводнике VS Code.
- Выберите Открыть структуру модели.
- Либо запустите эту команду из палитры команд и выберите файл.
- Используйте вкладки для просмотра архитектуры, модулей, тензоров и параметров.
- Выберите слой, чтобы открыть его свойства и встроенную справку.
Полное руководство по управлению, формулам, сравнению, экспорту и оценке операций открывается кнопкой Help на панели просмотрщика.
| |
