跳到正文
Hacker News· nityanandmathur·· 6 天前AI 评分40

TensorViz:在 VS Code 中探索 PyTorch 模型

Show HN: TensorViz – Explore PyTorch models inside VS Code

AI 导读

TensorViz 是一款可在 VS Code 中探索 PyTorch 模型的工具,提供 29 个交互式 PyTorch 架构,支持引导式浏览、查看捕获的 shape,并可将源码带回本地项目。无需安装、无需 Python 环境,每个模型都附带源码和一份小型复现配方,也可用 tensorViz 打开自己的 PyTorch 类。

正文 · 原文

MODEL EXPLORER / CURATED BY TENSORVIZ

Understand a model.
One layer at a time.

29 interactive PyTorch architectures. Follow a guided tour, inspect captured shapes, and take the source back to your own project.

No installation · No Python needed

CHOOSE A LEARNING PATH

A teaching order, not a claim that each later model adopted every earlier idea.

29 models to explore

BRING IT BACK TO YOUR PROJECT

Understand it here.
Keep exploring locally.

Every model includes its source and a small reproduction recipe. Open your own PyTorch class with tensorViz in VS Code.

Use tensorViz in VS Code

来源:Hacker News · tensorviz.dev